[leetcode] 【数组】15. 3Sum

最新推荐文章于 2024-09-07 16:05:05 发布

原创最新推荐文章于 2024-09-07 16:05:05 发布 · 245 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode

leetcode数组专栏收录该内容

24 篇文章

订阅专栏

本文探讨了如何寻找数组中三个整数相加等于零的所有唯一组合。通过使用排序和双指针技术来解决该问题，并实现了有效的去重措施。

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

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:
Elements in a triplet (a,b,c) must be in non-descending order. (ie, a ≤ b ≤ c)
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类似。方法也是那三种，a遍历3遍；b排序再夹逼；c遍历先取一个元素，剩下两个用two sum的标记法。
注意要采取去重措施。
这里我用的是方法b，先排序，再夹逼。

//cpp
class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int> > res;
        if(nums.size()<3)
            return res;
        sort(nums.begin(),nums.end());
        auto tail=nums.end();
        int target=0;
        for(auto first=nums.begin();first<nums.end()-2;first++)
        {
            if(*first==*(first-1)&&first!=nums.begin())
                continue;
            auto second=first+1;
            auto third=tail-1;
            while(second<third)
            {
                if(*first+*second+*third<target)
                    second++;
                else if(*first+*second+*third>target)
                    third--;
                else 
                {
                    res.push_back({*first,*second,*third});
                    second++;
                    third--;
                }
            }
        }
        sort(res.begin(),res.end());
        res.erase(unique(res.begin(),res.end()),res.end());
        return res;
    }
};