3Sum[leetcode]

最新推荐文章于 2020-06-30 10:31:49 发布

原创最新推荐文章于 2020-06-30 10:31:49 发布 · 519 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode

leetcode 专栏收录该内容

59 篇文章

订阅专栏

本文介绍了一种寻找数组中三个数相加等于特定目标值的算法实现，采用O(n^2)的时间复杂度，通过排序和双指针技巧避免了不必要的重复计算。此外，还详细解释了如何使用C++标准库函数来消除重复结果。

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

思路简单，O(n^2)复杂度

需要注意的是数组本身少于3个元素，以及重复答案的删除

vector<vector<int> > threeSum(vector<int> &num) {
        vector<vector<int> > res;
        if (num.size() <= 2) return res;
        sort(num.begin(), num.end());
        int twoSum;
        for (int i = 0; i < num.size() - 2; i++)
        {
            int l = i + 1, r = num.size() - 1;
            twoSum = 0 - num[i];
            while (l < r)
            {
                if (num[l] + num[r] < twoSum) l++;
                else if (num[l] + num[r] == twoSum)
                {
                    vector<int> three(3);
                    three[0] = num[i];
                    three[1] = num[l];
                    three[2] = num[r];
                    res.push_back(three);
                    l++;
                }
                else r--;
            }
        }
        sort(res.begin(),res.end());
        res.erase(unique(res.begin(), res.end()), res.end());
        return res;
    }

sort(res.begin(),res.end());
res.erase(unique(res.begin(), res.end()), res.end());

可以用于删除重复元素。unique函数将相邻的重复元素“删除”----其实是移到最后面，并且返回一个指向所有重复元素的迭代器。所以需要先sort一下再调用unique。之后利用erase去除重复元素