两数之和:
给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素不能使用两遍。
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
很多人第一反应就是用暴力破解,第一个循环控制每次取的第一个数,第二次循环控制第二个数,这样的时间复杂度为O(n^2)。
我们**使用一个哈希的结构来用空间换时间,使用map结构,map的key保存target-nums[i]的值,value保存当前下标i,**即当遍历到第一个nums数的2时,下标为0,用target值9-nums[0]=7,所以第一个key,value值为<7,0>,在下次遍历nums的其他元素时,看该元素是否在map中,在的话就返回对应value的值和当前的下标索引。
详细代码如下:
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
vector<int> result;
map<int, int> temp; // <key, value>
for(int i=0;i<nums.size();++i)
{
if(temp.count(nums[i]) != 0) //若map中有该值,直接取下map的下标
{
result.push_back(temp[nums[i]]);
result.push_back(i);
}
temp[target-nums[i]] = i; //保存<target-nums[i], i>
}
return result;
}
};
算法的时间复杂度为O(n)
如有更好的优化方案,欢迎批评指导。