Leetcode 15 - 3Sum（双指针）

题意

给一个数组，要求求数组里面所有的a, b, c，使得a + b + c = 0，要求解不重复。

思路

算法1

排序 + 二分。

先排序，然后枚举a和b，那么c = -a - b，然后二分查找一下c。

时间复杂度$O(n^2logn)$

注意去重：只要a和b不重复，那么c一定不会重复，于是枚举a和b时要跳过相同的值。

算法2

枚举 + Hash

我们用unordered_map存储每个元素出现的最后一个位置，然后枚举a和b，$O(1)$查找c是否出现并且下标大于a和b的下标。

时间复杂度$O(n^2)$。

算法3

枚举 + 2 Sum

先排序，然后枚举a，接下来只需要在a之后的位置找b + c = -a，即转化成了2 Sum。

解决2Sum问题时，我们用双指针指向数组的头和尾往中间走。

注意跳过重复元素。

代码

algorithm 1

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        int n = nums.size();
        vector<vector<int>> ans;
        if (n) {
            sort(nums.begin(), nums.end());
            for (int i = 0; i < n; i++) {
                for (int j = i + 1; j < n; j++) {
                    int x = 0 - nums[i] - nums[j];
                    int k = lower_bound(nums.begin() + j + 1, nums.end(), x) - nums.begin();
                    if (nums[k] == x && k < nums.size()) {
                        vector<int> tv;
                        tv.push_back(nums[i]);
                        tv.push_back(nums[j]);
                        tv.push_back(x);
                        ans.push_back(tv);
                    }
                    while (nums[j] == nums[j + 1]) j++;
                }
                while (nums[i] == nums[i + 1]) i++;
            }
        }
        return ans;
    }
};

algorithm 2

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        int n = nums.size();
        vector<vector<int>> ans;
        if (n) {
            sort(nums.begin(), nums.end());
            unordered_map<int, int> has;
            for (int i = 0; i < n; i++) {
                while (nums[i] == nums[i + 1] && i != n - 1) i++;
                has[nums[i]] = i;
            }
            for (int i = 0; i < n; i++) {
                for (int j = i + 1; j < n; j++) {
                    int x = 0 - nums[i] - nums[j];
                    if (has.count(x) && has[x] > j) {
                        vector<int> tv;
                        tv.push_back(nums[i]);
                        tv.push_back(nums[j]);
                        tv.push_back(x);
                        ans.push_back(tv);
                    }
                    while (nums[j] == nums[j + 1]) j++;
                }
                while (nums[i] == nums[i + 1]) i++;
            }
        }
        return ans;
    }
};

algorithm 3

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        int n = nums.size();
        vector<vector<int>> ans;
        if (n) {
            sort(nums.begin(), nums.end());
            for (int i = 0; i < n; i++) {
                int target = -nums[i];
                int j = i + 1, k = n - 1;
                while (j < k) {
                    if (nums[j] + nums[k] == target) {
                        vector<int> tv;
                        tv.push_back(nums[i]);
                        tv.push_back(nums[j]);
                        tv.push_back(nums[k]);
                        ans.push_back(tv);
                        while (j < k && nums[j] == tv[1]) j++;
                        while (j < k && nums[k] == tv[2]) k--;
                    } else {
                        if (nums[j] + nums[k] < target) j++;
                        else k--;
                    }
                }
                while (nums[i] == nums[i + 1] && i < n - 1) i++;
            }
        }
        return ans;
    }
};