LeetCode两数之和

不难写出一个o(n²)的解法

 vector<int> twoSum(vector<int>& nums, int target) {
        vector<int> res;
        for(int i=0;i<nums.size();i++){
            for(int j=i+1;j<nums.size();j++){
                if(nums[i]+nums[j]==target){
                    res.push_back(i);
                    res.push_back(j);
                }
            }
        }
        return res;
    }

注意到nums[i]+nums[j]==target,等式右端并没有变量,我们把nums[j]移过去, nums[i]==target-nums[j]
发现题目变成了,找到数组中的一个数字x,使得x满足 x=target-y,其中y是数组中的另一个数字
因此,我们可以先遍历一遍数组,把target-y放到哈希表中去,然后再次遍历,得到结果

      vector<int> twoSum(vector<int>& nums, int target) {
    	unordered_map<int,int> mp;
        vector<int> res;
        for(int i=0;i<nums.size();i++)
            mp[target-nums[i]]=i+1; 
            // 因为如果 mp没有k这个键,mp[k]的返回值就是0
            // 存入i+1是为了避免下标0和默认返回0冲突

        for(int i=0;i<nums.size();i++){
            if(mp[nums[i]]!=0&&mp[nums[i]]!=i+1){
                res.push_back(i);
                res.push_back(mp[nums[i]]-1);
                break;
            }
        }
        return res;
    }