【LeetCode】15. 三数之和

题目链接：https://leetcode-cn.com/problems/3sum/description/

题目描述

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

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

示例

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

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

解决方法

首先从小到大排序，然后遍历数组中每一个元素，用双指针找到另外两个元素，将三元组存入vector中（注意在此过程中要去重）

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
//         首先从小到大排序，然后遍历数组中每一个元素，用双指针找到另外两个元素,将三元组存入vector中（注意在此过程中要去重）
        
        vector<vector<int>> result;
        // 1.排序
        sort(nums.begin(),nums.end());
        
        // 2.找到三元组
        for (int i=0;i<nums.size();i++){
            if (nums[i]>0) break; //剪枝操作，如果当前数大于0则break
            if (i>0 && nums[i]==nums[i-1]) continue; //去重操作
            
            int left=i+1, right=nums.size()-1;
            while(left<right){
                int sum=nums[i]+nums[left]+nums[right];
                if (sum>0) right--; 
                else if (sum<0) left++;
                else{
                    result.push_back({nums[i],nums[left],nums[right]});
                    while (left<right && nums[left]==nums[left+1]) left++; //这两条语句是为了去重
                    while (left<right && nums[right]==nums[right-1]) right--;
                    left++;right--;
                }
            }
            
        }

        return result;
    }
};