题目链接:https://leetcode.cn/problems/two-sum/
这题其实不少做法,可以直接暴力O(n2)循环遍历但可能超时,也可以先排序后二分查找。但用hash会比较快。 hash可以自己实现,不过cpp的unordered_map可以代劳。
先复习一下unordered_map(参考【C++】unordered_map 的使用_生活不允许普通人内向的博客-CSDN博客):
1. unordered_map是STL的一种容器,内部采用hash表实现
2. 特点:无序(不会对存储的键值对进行排序),每个值对应一个键,不存在两个元素的键一样
3. 检索快速,find()时间复杂度为O(1),本题也是基于这个特点
4. unordered_map的查找、插入、删除的时间复杂度都是O(1),但map是O(logN)
具体的实现代码就非常简单了,就是遍历所有元素,如果当前元素可以在unordered_map中找到一个与其相加==target的值,就返回;否则就把这个元素加入unordered_map,用作对后续新添加的元素的匹配。唯一的难点在于hashtable中的键是元素值,值才是元素下标,有点绕(因为find函数是查找键是否存在的,存元素下标没有意义)
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
unordered_map<int,int> hashtable;
for(int i=0;i<nums.size();i++){
auto it=hashtable.find(target-nums[i]);
if(it!=hashtable.end()) return {i,hashtable[target-nums[i]]};
hashtable[nums[i]]=i; //find是找是否存在这个键,因此键存的应该是nums[i]的值
}
return {};
}
};