2021-11-11剑指offerii007.数组中和为0的三个数

在这里插入图片描述

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        if (nums.size() < 3) return {};
        vector<vector<int>> res;

        sort(nums.begin(), nums.end());
        for (int i = 0; i < nums.size() - 2; i++) {
            if (i > 0 && nums[i] == nums[i - 1]) continue;

            // 在 i + 1 ... nums.length - 1 中查找相加等于 -nums[i] 的两个数
            int target = - nums[i];
            int left = i + 1, right = nums.size() - 1;
            while (left < right) {
                int sum = nums[left] + nums[right];
                if (sum == target) {
                    res.push_back({nums[i], nums[left], nums[right]});
                    /*
                        下面的代码相当于:
                        while (left < right) {
                            // 不管前后相不相等,left 都要往前走
                            left++;
                            if (nums[left - 1] != nums[left]) break;
                        }
                        while (left < right) {
                            // 不管前后相不相等,right 都要往后走
                            right--;
                            if (nums[right + 1] != nums[right]) break;
                        }
                     */
                    // 去重
                    while (left < right && nums[left] == nums[++left]);
                    while (left < right && nums[right] == nums[--right]);
                } else if (sum < target) {
                    left++;
                } else {
                    right--;
                }
            }
        }
        return res;
    }
};

作者:tangweiqun
链接:https://leetcode-cn.com/problems/1fGaJU/solution/jian-dan-yi-dong-javac-pythonjs-san-shu-nu6el/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。

本菜鸡写的没有去重功能的代码

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> ret;
        if(nums.size() < 3)
        return ret;
        sort(nums.begin(),nums.end());
        for(int i = 0;i < nums.size()-2;i++){
            int left = i + 1;
            int right = nums.size()-1;
            while(left < right){
                //int mid = left + (right - left)/2;
                int sum = nums[i] + nums[left] + nums[right];
                if(sum == 0){
                    ret.push_back({nums[i],nums[left],nums[right]});
                    // ret.push_back();
                    // ret.push_back();
                    //找到一组之后应该如何处理呢?
                    //又该如何去重呢?
                    left++;
                    
                }
                else if(sum > 0){
                    right--;
                }
                else if(sum < 0){
                    left++;
                }
            }
        }
        return ret;
    }
};

关于去重有3个部分:
1)每次寻找一个起点的时候可能数字会重复

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值