力扣1.两数相加

两数相加题目

思路

暴力枚举的话时间复杂度是$O(n²)$，既然大量时间都花费在寻找target - x上，那么就要降低查找的时间。使用unordered_map可以将查找的时间复杂度降到$O(1)$，这样的话只需遍历一次数组，时间复杂度就可以降到$O(n)$。

代码

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        unordered_map<int, int> nums_map;
        // 将数组的数字和下标录入map
        for (int i = 0; i < nums.size(); i++)
            nums_map.emplace(nums[i], i);
        // 遍历查找
        for (int x : nums) {
            if (nums_map.find(target - x) != nums_map.end())
                return {nums_map[x], nums_map[target - x]};
        }
        return {};
    }
};

但是这个出现了错误解答。
错误在于可能会一个元素重复出现，如[3, 2, 4], 6 就会输出[0, 0]。
为了避免这个问题，需要在每次查找target - x时都不要查找x本身。那么如何在map里不查找一个键呢？只要这个map里没有这个键，自然就不会查找到了。
因此可以改变下思路，将键值对的插入动作放到判断相加之和的后面。即：先遍历数组判断有无target - x，如果没有再将其插入map。

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        unordered_map<int, int> nums_map;
        for (int i = 0; i < nums.size(); i++) {
            if (nums_map.find(target - nums[i]) != nums_map.end())
                return {i, nums_map[target - nums[i]]};
            nums_map.emplace(nums[i], i);
        }
        return {};
    }
};