【Leetcode】15.三数之和 【排序+双指针】

给定一个包含 $n$ 个整数的数组 $nums$，判断 $nums$ 中是否存在三个元素 $a，b，c $，使得 $a+b+c=0？$找出所有满足条件且不重复的三元组。

注意：答案中不可以包含重复的三元组。

例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4]，

满足要求的三元组集合为：
[
  [-1, 0, 1],
  [-1, -1, 2]
]

解：求解两数之和：固定一个值，用$map$搜索另一个值；那么类似于这种做法，我们也可以先固定一个值，然后双指针进行类似于二分搜索操作，中间包含去重操作【排序是为了简化问题】

class Solution {
public:
    /*
     *  固定一个target，以双指针遍历寻找两数之和等于target
     *  不断向内压缩，l与r不断接近
     */
    vector<vector<int>> threeSum(vector<int>& nums) {
        sort(nums.begin(), nums.end());  //排序
        int n = nums.size();
        vector<vector<int>> vec;
        for (int i = 0; i < n - 2; ++i) {
            //排除重复值
            if (i == 0 || i > 0 && nums[i] != nums[i - 1]) {
                int l = i + 1, r = n - 1;
                while (l < r) {
                    int sum = nums[i] + nums[l] + nums[r];
                    if (sum > 0) {
                        while (l < r && nums[r] == nums[r - 1]) r--;  //去重
                        r--;
                    }
                    else if (sum < 0) {
                        while (l < r && nums[l] == nums[l + 1]) l++;
                        l++;
                    }
                    else {
                        vec.push_back({nums[i], nums[l], nums[r]});
                        while (l < r && nums[l] == nums[l + 1]) l++;
                        while (l < r && nums[r] == nums[r - 1]) r--;
                        l++;
                        r--;
                    }
                }
            }
        }
        return vec;
    }
};