NC54—数组中相加和为0的三元组

题意

给出一个有n个元素的数组S，S中是否有元素a, b, c满足a + b + c = 0？找出数组S中所有满足条件的三元组

注意：

三元组（a、b、c）中的元素必须按非降序排列（即a ≤ b ≤ c）

解集中不能包含重复的三元组。

例如，给定的数组 S = {-10 0 10 20 -10 -40}, 解集为(-10, 0, 10) (-10, -10, 20)

题解

先排序，然后以第一个值为基准开始遍历，用双指针求第二个值和第三个值

class Solution {
public:
    vector<vector<int> > threeSum(vector<int> &num) {
        sort(num.begin(), num.end());
        vector<vector<int> > res;
        if (num.size() < 3) return res;
        for (int i = 0; i < num.size() - 2; ++i) {
            int j = i + 1;
            int k = num.size() - 1;
            int target = -num[i];
            while (j < k) {
                if (num[j] + num[k] > target) --k;
                else if (num[j] + num[k] < target) ++j;
                else {
                    vector<int> current = {num[i], num[j], num[k]};
                    res.push_back(current);
                    while (j + 1 < k && num[j+1] == num[j]) ++j;  // 防止重复
                    while (k - 1 > j && num[k-1] == num[k]) --k;  // 防止重复
                    ++j; --k;
                }
            }
            while (i + 1 < num.size() - 2 && num[i+1] == num[i]) ++i;   // 防止重复
        }
        return res;
    }
};