【LeetCode-1】两数之和

问题

给定一个整数数组 nums 和一个整数目标值 target，请你在该数组中找出 和为目标值 的那 两个 整数，并返回它们的数组下标。

你可以假设每种输入只会对应一个答案。但是，数组中同一个元素在答案里不能重复出现。

你可以按任意顺序返回答案。

示例

输入： nums = [2,7,11,15], target = 9
输出： [0,1]
解释： 因为 nums[0] + nums[1] == 9 ，返回 [0, 1] 。

解答1：哈希表（O(n)）

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        unordered_map<int, bool> ump;
        for (int i = 0; i < nums.size(); i++) {
            int cur = target - nums[i];
            if (ump.count(cur)) return vector<int>{i, ump[cur]};
            ump[nums[i]] = i;
        }
        return vector<int>{-1, -1};
    }
};

解答2：通用解法（O(nlogn）

class Solution {
public:
    vector<vector<int>> twoSum(vector<int>& nums, int target) {
        sort(nums.begin(), nums.end());
        int lo = 0, hi = nums.size() - 1;
        vector<vector<int>> res;
        while (lo < hi) {
            int left = nums[lo], right = nums[hi];
            int sum = left + right;
            if (sum < target)
                while (lo < hi && nums[lo] == left) lo++;
            else if (sum > target)
                while (lo < hi && nums[hi] == right) hi--;
            else {
                res.push_back({left, right});
                while (lo < hi && nums[lo] == left) lo++;
                while (lo < hi && nums[hi] == right) hi--;
            }
        }
        return res;
    }
};

重点思路

如果把题目要求改为数字可能有重复，答案不止一个，要求输出所有可能的结果（输出数值而不是角标，否则不能用排序），则可使用排序+双指针这样的通用解法，这一解法能服用在3Sum和nSum问题上，需要重点掌握。