原题:力扣(LeetCode)官网 - 全球极客挚爱的技术成长平台https://leetcode.cn/problems/two-sum/submissions/
为什么这题可以用哈希表来解?
当查找一个元素是否在一个集合里出现过的时候,首先可以考虑用哈希法。
题目中我们可以通过target减去集合中的元素,判断得到的值是否在集合中出现过,所以用哈希表。
为什么用map呢?
因为题目要求返回下标,用map最方便。
为什么用unordered_map?
因为unordered_map底层实现是哈希表,题目没有要求结果要排序,仅仅是查找元素的话,unordered_map效率最高。
注意本题往哈希表插入元素时,把元素原来集合的下标也放进哈希表中,不要用哈希表默认的下标值,否则输出错误。
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
unordered_map<int,int>mp;
for(int i=0;i<nums.size();i++){
auto iter=mp.find(target-nums[i]);
if(iter!=mp.end()){
return {i,iter->second};
}
mp.insert(pair<int,int>{nums[i],i});
}
return {};
}
};