数组中和为0的三元组

数组中相加和为0的三元组

http://www.nowcoder.com/questionTerminal/345e2ed5f81d4017bbb8cc6055b0b711

先排序,然后以第一个值为基准开始遍历,用双指针求第二个值和第三个值。

class Solution {
public:
    vector<vector<int> > threeSum(vector<int> &num) {
        sort(num.begin(), num.end());
        vector<vector<int> > res;
        if (num.size() < 3) return res;
        for (int i = 0; i < num.size() - 2; ++i) {
            int j = i + 1;
            int k = num.size() - 1;
            int target = -num[i];
            while (j < k) {
                if (num[j] + num[k] > target) --k;
                else if (num[j] + num[k] < target) ++j;
                else {
                    vector<int> current = {num[i], num[j], num[k]};
                    res.push_back(current);
                    while (j + 1 < k && num[j+1] == num[j]) ++j;  // 防止重复
                    while (k - 1 > j && num[k-1] == num[k]) --k;  // 防止重复
                    ++j; --k;
                }
            }
            while (i + 1 < num.size() - 2 && num[i+1] == num[i]) ++i;   // 防止重复
        }
        return res;
    }
};
刷遍天下无敌手 文章被收录于专栏

秋招刷题历程

全部评论
题目说数组一定是按顺序排的了吗
点赞 回复 分享
发布于 2021-03-15 10:49
感觉第22行没必要写 i + 1 < num.size() 吧 for里面已经限定了i的范围;而且i也不会数组越界
点赞 回复 分享
发布于 2021-04-05 12:48

相关推荐

神哥了不得:你简历字体有点不太协调呀,下面的字实在太小了呀,而且项目也不太行,建议换几个高质量的项目,面试会多很多
点赞 评论 收藏
分享
小狗吃臭臭:以后用不到你设计的手机了,可惜!
点赞 评论 收藏
分享
评论
15
收藏
分享

创作者周榜

更多
牛客网
牛客企业服务