题目
给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
经验总结:
可以使用unordered_map的键(key)来存储数组中每个数,值(value)来存储这个数字在数组的下标。然后遍历数组的时候查询target减去此数字后的差是否存在unordered_map中。
记:unordered_map比map快(不排序的话)。
首次提交出现了错误
然后发现是使用了同一个元素导致的,修改后AC。
C++代码:
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
unordered_map<int,int> um;
for(int i=0;i<nums.size();i++){
um.insert(make_pair(nums[i],i));
}
for(int i=0;i<nums.size();i++){
if(um.find(target-nums[i]) != um.end()){
if(i == um[target-nums[i]]) continue; //注意需要跳过同一元素
return {i,um[target-nums[i]]};
}
}
return {1};
}
};
好奇心带着我去看了下耗时比我少的代码,贴一下别人家的代码。
并引用下题解中map的解释:
思路:采取一边插入哈希表一边寻找一边在已经插入的哈希表中寻找的方式,每次都拿着即将插入哈希表的数字然后在哈希表中找是否存在剩下的那个函数。有人可能会问,在循环里面查找的话时间复杂度不就变高了嘛,但是哈希表查找时间可以说是o(1)是常数(避免冲突的算法采用链地址可能不为o(1),但一般不用链地址来避免冲突),所以总的时间复杂度就为o(n),哈希表就是一种用空间换取时间的算法。