难度:简单
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/two-sum
1. 问题描述
给定一个整数数组nums
和一个整数目标值target
,请你在该数组中找出 和为目标值target
的那 两个整数,并返回它们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。你可以按任意顺序返回答案。
2. 分析与解答
本题被标注为简单题,对于本题,是需要找到两个数,最简单的分析就是有两个指针不断循环遍历数组,如下图所示:
设置指针i指向当前值,指针j指向始终从指针i的下一个位置开始,这样需要双重循环,时间复杂度为O(n2)。
要想减少时间复杂度,可以利用空间换时间,可以将已经遍历过的存储下来并且方便查询,可以使用map
存储已经遍历的数,其中,key为数值,value为在原数组中的位置。对于当前遍历的数nums[i],在map
中查找target-nums[i]。
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
int n = nums.size();
map<int, int> mp;
vector<int> res;
for (int i = 0; i < n; i++) {
if (i > 0 && mp.find(target-nums[i]) != mp.end()) {
res.push_back(i);
res.push_back(mp[target-nums[i]]);
break;
} else {
mp[nums[i]] = i;
}
}
return res;
}
};
这样时间复杂度为O(n)。
发散:如果本题中的数组是排好序的,如示例1,那么可以使用头尾双指针,这样只需要一次遍历,即如下图所示:
如果当前指针i和指针j的和等于target,则返回i和j的值;如果当前指针i和指针j的和小于target,则i++;否则j–。在这里要返回位置,如果先排序将破坏原来的顺序,还得存储原先的位置信息,不如上述的map
方案。