LeetCode 15.三数之和

15. 三数之和

medium

题目描述：

给你一个整数数组 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 。

解题思路：

1. 获取输入数组的大小，并赋值给一个整数变量size。如果size小于3，说明无法找到三元组，直接返回一个空的二维向量。
2. 定义一个空的二维向量res，用于存储结果。
3. 对输入数组nums进行升序排序，以便使用双指针法。
4. 使用一个for循环，从第一个元素开始遍历数组nums，每次循环中，将当前元素赋值给一个整数变量i。循环的终止条件是i达到size。
5. 如果当前元素nums[i]大于零，说明后面的元素都大于零，无法和它们组成和为零的三元组，直接返回res。
6. 如果当前元素nums[i]和前一个元素nums[i-1]相等（注意要排除i等于0的情况），说明这个元素已经被处理过了，为了避免重复结果，跳过这次循环，继续下一次循环。
7. 定义两个整数变量left和right，分别表示左指针和右指针，初始时分别指向i的下一个位置和数组的最后一个位置。
8. 使用一个while循环，在left小于right的条件下，执行以下操作：
   • 如果nums[left] + nums[right]大于-nums[i]，说明和太大了，需要减小右指针right，使其指向前一个位置。
   • 如果nums[left] + nums[right]小于-nums[i]，说明和太小了，需要增大左指针left，使其指向后一个位置。
   • 如果nums[left] + nums[right]等于-nums[i]，说明找到了一个符合条件的三元组，将它们放入一个一维向量中，并将这个一维向量添加到res中。然后同时增大左指针left和减小右指针right，并跳过所有重复的元素（注意要保证left小于right），以便寻找下一个可能的三元组。
9. 当for循环结束时，返回res作为最终结果。

代码实现：

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        int size = nums.size(); //获取输入数组的大小
        if (size < 3) //如果大小小于3，无法找到三元组
            return {}; //返回空结果
        vector<vector<int>> res; //定义空结果
        sort(nums.begin(), nums.end()); //对输入数组进行升序排序
        for (int i = 0; i < size; i++) { //从第一个元素开始遍历数组
            if (nums[i] > 0) { //如果当前元素大于零
                return res; //直接返回结果
            }
            if (i > 0 && nums[i] == nums[i - 1]) { //如果当前元素和前一个元素相等
                continue; //跳过这次循环
            }
            int left = i + 1; //定义左指针
            int right = size - 1; //定义右指针
            while (left < right) { //当左指针小于右指针时
                if (nums[left] + nums[right]+nums[i] > 0) { //如果和大于零
                    right--; //减小右指针
                }
                else if(nums[left] + nums[right]+nums[i]<0 ){ //如果和小于零
                    left++; //增大左指针
                }
                else { //如果和等于零
                    res.push_back(vector<int>{nums[i], nums[left], nums[right]}); //添加三元组到结果中
                    left++; //增大左指针
                    right--; //减小右指针
                    while (left < right && nums[left] == nums[left - 1]) { //跳过重复的左元素
                        left++;
                    }
                    while (left < right && nums[right] == nums[right + 1]) { //跳过重复的右元素
                        right--;
                    }
                }
            }
        }
        return res; //返回最终结果
    }
};