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;
    }
};```