Day 4 三数之和

三数之和

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

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

示例：

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

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

使用语言：C++

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
            vector<vector<int>> solu;
            int n=nums.size(),l,r,sum;
            if (n <= 2) return solu;
            sort(nums.begin(),nums.end());
            for(int i=0;i<n-2&&nums[i]<=0;i++){
                l=i+1;
                r=n-1;
                while(l<r){
                     sum=nums[i]+nums[l]+nums[r];
                   if(sum==0) {	 
                       vector<int> temp={nums[i],nums[l],nums[r]};
                       solu.push_back(temp);
                       while(l<r&&nums[l]==nums[l+1]) l++;
                       while(l<r&&nums[r]==nums[r-1]) r--;
                       l++;
                       r--;
                        }
                   else if(sum<0) l++;
                   else if(sum>0) r--;
                }
                while(i+1<n-2&&nums[i]==nums[i+1]) i++;
            }
            return solu;
    }
};
 

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