leetcode-前300经典刷题-15（C++）

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

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/3sum
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

本题的难点在于时间复杂度的控制（优化算法）和 （去重）
思路： 控制变量的方式减少运算+排序优化

class Solution {
public:
    vector> threeSum(vector& nums) 
    {
        // 类似控制变量法，同时排序
        // 本题的难点是去重
        vector> res;
        // special input 
        if (nums.size() <= 2) {
            return res;
        }
        if (nums.size() == 3) {
            if (nums[0] + nums[1] + nums[2] == 0) {
                res.push_back({nums[0], nums[1], nums[2]});
            }
            return res;
        }
        // 多个元素
        // 排序
        sort(nums.begin(), nums.end());
        for (int i = 0; i < nums.size()-2; i++) {
            // 去重1
            if (i > 0 && nums[i] == nums[i - 1]) {
                continue;
            }
            // 排序后，正数之后不可能出现和为0
            if (nums[i] > 0) {
                break;
            }
            // 二分
            int l = i + 1;
            int r = nums.size() - 1;
            while (l < r) {
                int sum = nums[i] + nums[l] + nums[r];
                if(sum == 0) {
                    // 去重
                    while (l < r && nums[l] == nums[l + 1]) {
                        l++;
                    }
                    // 去重
                    while (l < r && nums[r] == nums[r - 1]) {
                        r--;
                    }
                    res.push_back({nums[i], nums[l], nums[r]});
                    l++;
                    r--;
                } else if (sum < 0) {
                    l++;
                } else {
                    r--;
                }

            }
        }
        return res;
    }
};