lc.15.三数之和

leetcode. 15.三数之和

给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有和为 0 且不重复的三元组。

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

示例 1:

输入:nums = [-1,0,1,2,-1,-4] 输出:[[-1,-1,2],[-1,0,1]]

示例 2:

输入:nums = [] 输出:[]

示例 3:

输入:nums = [0] 输出:[]

提示:

0 <= nums.length <= 3000 -105 <= nums[i] <= 105

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        sort(nums.begin(),nums.end() );
        int n = nums.size();
        
        vector< vector<int >> res;
        for(int i=0;i<n;i++) {
            if(nums[i] > 0) break;
            if (i>0 &&nums[i] == nums[i-1]) continue;
            int a= nums[i];

            
            int l = i+1,r = n-1;
            while(l <r ) {
                int b = nums[l],c = nums[r];
                if (a+b+c == 0) {
                    res.push_back({a,b,c});
                    while(l<r && nums[l] == nums[l+1]) ++l;
                    while(l<r && nums[r] == nums[r-1]) --r;
                    
                    
                    ++l;
                }else if (a+b+c>0) {

                    --r;

                }else {
                    //<0
                    ++l;
                }
            }
        }
        return res;
    }
};