[M双指针] lc15. 三数之和(双指针+去重判断+经典)

1. 题目来源

链接：lc15. 三数之和

2. 题目解析

方法一：枚举+双指针

前导题：

• [E哈希] lc1. 两数之和(哈希+二分+双指针)

且本题与后面几个题都强相关。

双指针思路：

• 先枚举一个数，记为指针 i。再进行双指针，j 在 i 的后一位， k 在数组末尾。
• 排序后。三元组成立时，即有：num[i]+num[j]+num[k] == 0 则找到三元组 {i, j, k}。若发现 num[i]+num[j]+num[k] > 0，则 k -- 指针向左走变小。否则，j ++ 向右走变大。
• 所以，针对每一个 num[i] 就可以用 $O(n)$ 的时间找到合法的 j, k。 故总的时间复杂度就是 $O(n^2)$ 的。

关键是去重：

• 若 i 与上一个枚举的 i 值相同则跳过，因为针对这个相同数，所有三元组情况已经在上一次枚举过了。故可以避免 i 的重复且不遗漏。
• 同理 j 也是，若和上一个枚举的 j 相同，则跳过。注意 j>i+1 判重时候的代码细节。
• 对于 k 来讲，每次选择 i, j 作为二元组下的最左边的 k 使用即可。其实，i， j 不重复的话，k 完全由 i, j 决定，故重复与否是无关紧要的。k 不断的向左试探，判断 k-1 能否转移过去，尽量向左走，在这个过程中，也会跳过相同的 k 值，也做到了 k 的去重。

双指针优化其实不难。重点还是在了判重上面。针对 1 1 2 2 -3 -3 这个就能表示 i, j 都要判重的情况了。也能表示 k 一直向左试探下一个数的情况了。

时间复杂度：$O(n^2)$
空间复杂度：$O(1)$

代码：

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