其他分享
首页 > 其他分享> > Offer II 007. 数组中和为 0 的三个数

Offer II 007. 数组中和为 0 的三个数

作者:互联网

题目描述

给定一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a ,b ,c ,使得 a + b + c = 0 ?请找出所有和为 0 且 不重复 的三元组。
0 <= nums.length <= 3000
-105 <= nums[i] <= 105

解题思路

(参考006)

a+b+c = 0 转化为 a+b = -c ,即对于每个常数c,在数组中找到两个不同的数a,b,使得a+b =c。 用sort将向量升序排列,c<=b<=a,选取不大于0的不同元素作为c,在目标元素的后面用双指针找b和a,注意去重即可

代码

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> result;
        vector<int> test(3);
        sort(nums.begin(),nums.end());
        for(int i=0;i<nums.size();i++){
            if(i>=1 && nums[i-1] == nums[i]){
                continue;
            }
            if(nums[i]>0) break;
            int prel = -10001,prer=-10001;
            int left = i+1, right=nums.size()-1;
            while(left<right){
                if(nums[left] == prel){
                    left++;
                    continue;
                }
                if(nums[right] == prer){
                    right--;
                    continue;
                }
                if(nums[left]+nums[right] == -1*nums[i]){
                    cout<<"("<<i<<","<<left<<","<<right<<") ";
                    prel = nums[left];
                    prer = nums[right];
                    test[0] = nums[i];
                    test[1] = nums[left];
                    test[2] = nums[right];
                    result.push_back(test);
                    }
                    left++;
                    right--;
                }
                else if(nums[left]+nums[right] > -1*nums[i]){
                    right--;
                }
                else left++;
            }
        }

        return result;
    }
};```

标签:right,continue,nums,int,Offer,II,vector,007,left
来源: https://blog.csdn.net/Matsumea/article/details/123205952