链接
我的代码:
class Solution {
public:vector<int> twoSum(vector<int>& nums, int target) {unordered_map<int,int> hash;vector<int>res;for(int i = 0;i<nums.size();i++){hash[nums[i]] = i;}for(int i = 0;i<nums.size();i++){if(hash.count(target-nums[i])&&hash[target-nums[i]]!=i){//注意加hash[target-nums[i]]!=i目的是防止出现6=3+3这种caseres.push_back(i);res.push_back(hash[target-nums[i]]);break;}}return res;}
};
更好的代码:
class Solution {
public:vector<int> twoSum(vector<int>& nums, int target) {unordered_map<int,int> hash;for(int i = 0;i<nums.size();i++){if(hash.count(target-nums[i])){return {hash[target-nums[i]],i};}hash[nums[i]] = i;}return {};}
};
也就是一边添加,一边判断,因为题目保证了肯定存在两个不重复元素的和是target,所以遍历一遍之后肯定可以找到。