首页 > 代码库 > 15. 3Sum

15. 3Sum

2024-09-12 09:54:39   214人阅读

Given an array S of n integers, are there elements a, b, c in S such that a + b + c = 0? Find all unique triplets in the array which gives the sum of zero.

Note: The solution set must not contain duplicate triplets.

For example, given array S = [-1, 0, 1, 2, -1, -4],

A solution set is:
[
  [-1, 0, 1],
  [-1, -1, 2]
]
class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        sort(nums.begin(),nums.end());
        vector<vector<int>>ans;
        int n=nums.size();
        for(int i=0;i<n-2;i++){
            if(i&&nums[i]==nums[i-1])continue;//跳过重复值
            int low=i+1,high=n-1,mid,sum=-nums[i];//Two-Sum problem
            while(low<high){
                if(nums[low]+nums[high]==sum){
                    ans.push_back({nums[i],nums[low],nums[high]});//find one
                    while(low<high&&nums[low]==nums[low+1])low++;
                    while(low<high&&nums[high]==nums[high-1])high--;
                    low++,high--;
                }
                else if(nums[low]+nums[high]<sum){
                    while(low<high&&nums[low]==nums[low+1])low++;
                    low++;
                }
                else {
                      while(low<high&&nums[high]==nums[high-1])high--;
                      high--;
                }
            }
        }
        return ans;
    }
};

 

15. 3Sum


联系
我们