三数之和（双指针解法）

今天继续看双指针专项。

一、题目

给你一个整数数组 nums ，判断是否存在三元组 [nums[i], nums[j], nums[k]] 满足 i != j、i != k 且 j != k ，同时还满足 nums[i] + nums[j] + nums[k] == 0 。请

你返回所有和为 0 且不重复的三元组。

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

示例 1：

输入：nums = [-1,0,1,2,-1,-4]
输出：[[-1,-1,2],[-1,0,1]]
解释：
nums[0] + nums[1] + nums[2] = (-1) + 0 + 1 = 0 。
nums[1] + nums[2] + nums[4] = 0 + 1 + (-1) = 0 。
nums[0] + nums[3] + nums[4] = (-1) + 2 + (-1) = 0 。
不同的三元组是 [-1,0,1] 和 [-1,-1,2] 。
注意，输出的顺序和三元组的顺序并不重要。

示例 2：

输入：nums = [0,1,1]
输出：[]
解释：唯一可能的三元组和不为 0 。

示例 3：

输入：nums = [0,0,0]
输出：[[0,0,0]]
解释：唯一可能的三元组和为 0 。

二、代码（来自天问AI解法）

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> result;//用于存储结果
        int n = nums.size();
        sort(nums.begin(),nums.end());//对数组元素进行排序
        for(int i = 0 ; i < n-2 ; i++){
            if(i>0 && nums[i] == nums[i-1]){
               continue;
            }
            int j = i + 1;
            int k = n - 1;
            while(j<k){
            int sum = nums[i] + nums[j] + nums[k];
            if(sum>0){
                k--;
            }
            else if(sum<0){
                j++;
            }
            else{
            result.push_back({nums[i],nums[j],nums[k]});
            while(j<k && nums[j]==nums[j+1])j++;
            while(j<k && nums[k]==nums[k-1])k--;
            j++;
            k--;
            }
            

            }
  


        }
        return result;

    }
};

三、理解及解析

来看各段代码

vector<vector<int>> result;//用于存储结果
        int n = nums.size();
        sort(nums.begin(),nums.end());//对数组元素进行排序

开始时，初始化了一个动态数组，用于存储每一个和为零的三元数组，并且对传入动态数组进行排序。

        for(int i = 0 ; i < n-2 ; i++){
            if(i>0 && nums[i] == nums[i-1]){
               continue;
            }

进入for循环，因数组中数据是递增的，而且题目中要求  输入不重复的三元组，因此，在如果存在重复的元素，则跳过该元素，例如[-1,-1,-1,0,1,2],我们只需要从最后一个 -1 开始遍历就行，因此当nums[i] == nums[i-1]时，不做什操作，i++后进行下一次循环。至于为什么是 n-2,因为该大循环是固定i后，对于每个i，从i后边的剩下元素中去寻找和为零的情况，至少有三个数相加，当i=n-2时，说明此时i以前的所有元素均已经遍历完，此时还剩余一个元素，不满足三个元素。

 int j = i + 1;
            int k = n - 1;
            while(j<k){
            int sum = nums[i] + nums[j] + nums[k];
            if(sum>0){
                k--;
            }
            else if(sum<0){
                j++;
            }

对于循环中的每个i，j指向i后的第一个元素，k指向数组尾部元素，固定i，后 j 和 k 开始向中间开始循环遍历，对于每次的遍历，根据和sum的大小来判断指针的移动。

一、若sum>0，说明加多了，需要换个小点的数，sum才能为零，j 和 k 只能向中间走，又因为数组递增，因此 k--。

二、若sum<0, 说明相加元素中负数需要增大些，因此j++.

三、sum=0,此时需要将三个数组成数组加入result.同时向中间移动两指针。即为接下来这段代码

 else{
            result.push_back({nums[i],nums[j],nums[k]});
            while(j<k && nums[j]==nums[j+1])j++;
            while(j<k && nums[k]==nums[k-1])k--;
            j++;
            k--;

在将三元素加入后，因为指针 j 的前面和 k的后面可能存在重复元素，因此可采用while 循环跳过与当前一样的元素。

最后返回结果result。