【LeetCode】三数之和

题目

【中等】三数之和
描述
给出一个有n个元素的数组S，S中是否有元素a,b,c满足a+b+c=0？找出数组S中所有满足条件的三元组。

数据范围：0 \le n \le 10000≤n≤1000，数组中各个元素值满足 |val | \le 100∣val∣≤100
空间复杂度：O(n^2)
)，时间复杂度 O(n^2)

注意：
三元组（a、b、c）中的元素必须按非降序排列。（即a≤b≤c）
解集中不能包含重复的三元组。

思路

使用双指针：

1. 先将数组原地排序
2. 遍历数组，每遍历到一个元素时：
   1. 设置左指针和右指针
   2. 若当前三数之和为0，则加入结果
   3. 若三数之和大于0，右指针–
   4. 若三叔之和小于0，左指针++

代码

class Solution {
public:
    vector<vector<int> > threeSum(vector<int> &num) {
        vector<vector<int>> res;
        if(num.size() < 3) return res; //输入检验

        sort(num.begin(), num.end());
        for(int i = 0; i < num.size() - 2; i++){
            if(i>0 && num[i] == num[i-1]) continue;

            int left = i+  1;
            int right = num.size() - 1; 
            while(left < right){
                if(num[i] + num[left] + num[right] == 0){
                    res.push_back({num[i], num[left], num[right]});
                    while(num[left] == num[left+1]) left++;
                    while(num[right] == num[right]-1) right--;
                    left++;
                    right--;
                }
                else if(num[i] + num[left] + num[right] > 0){
                    right--;
                }
                else if(num[i] + num[left] + num[right] < 0){
                    left++;
                }
            }
        }
        return res;
    }
};

笔记