leetcode(M) three sum

最新推荐文章于 2022-05-18 20:31:27 发布

Jang1996

最新推荐文章于 2022-05-18 20:31:27 发布

阅读量404

点赞数

分类专栏： LeetCode

LeetCode 专栏收录该内容

36 篇文章

订阅专栏

这是一个关于LeetCode中三数之和问题的讨论。文章指出，给定一个整数数组S，目标是找到所有和为0的不重复三元组。作者提出了将问题转化为两数之和的初始思路，并分享了相关代码，但注意到该方法无法排除重复的三元组。随后，文章提及了一种优化策略，即在遍历过程中使用集合来避免重复，但未给出具体实现。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

3Sum QuestionEditorial Solution My Submissions
Total Accepted: 159017
Total Submissions: 778216
Difficulty: Medium
Contributors: Admin
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]
]
最初思路：将问题转化成two sum
代码：

class Solution {
public:
//two sum
vector<int> twoSum(vector<int>& nums,int target) {
        map<int,int>m;
        for(int i=0;i<nums.size();++i)
        if(m.count(target-nums[i]))
        {if(m[target-nums[i]]!=i)
        return {nums[i],target-nums[i],-target};}
        else m[nums[i]]=i;
        return {};
    }
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> v;
        for(int i=0;i<nums.size();++i){

          v.push_back(  twoSum(nums,-nums[i]));
        }
        return v;
    }
};

但是此代码不能将重复的去除，结果如下：

Input:
[-1,0,1,2,-1,-4]
Output:
[[1,0,-1],[1,-1,0],[0,-1,1],[-1,-1,2],[1,0,-1],[]]
Expected:
[[-1,-1,2],[-1,0,1]]

如果要改进的话。可以在twosum里将那三个数排序，然后在threesum里用set这种不重复容器存储，这里我就不改进了。
第二种解法：

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
       set<vector<int>> s;int j,k;
       sort(nums.begin(),nums.end());
       for(int i=0;i<nums.size();++i){
           j=i+1;k=nums.size()-1;
           while(j<k){
               if(nums[i]+nums[j]+nums[k]==0)
             {  s.insert({nums[i],nums[j],nums[k]});
                while(j<k&&nums[k]==nums[k-1]) --k;//为了避免重复
                while(j<k&&nums[j]==nums[j+1]) ++j;//为了避免重复
                //这里是不需要再对nums[i]==nums[i+1]再做考虑，因为set不允许重复
             ++j;--k;}
               else
               if(nums[i]+nums[j]<-nums[k]) ++j;
               else --k;
           }
       }
       return{s.begin(),s.end()};
    }
};