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

题目

力扣

思路

排序+双指针。先判断数组的大小，如果小于3就返回空数组。首先对数组进行排序，确定第一个元素下标为k，如果这个第一个元素大于0就退出，再用双指针遍历另外后两个元素与第一个元素的和是否为0.难点在于如何“不重复”，解决思路是在找到一个三元组后，重新修改i和j时注意修改后的值和修改前的不同。

代码

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        int n=nums.size();
        if(n<3) return {};
        vector<vector<int>> ans;
        sort(nums.begin(),nums.end());
        int k=0;
        while(k<n-2){
            int t=nums[k];
            if(t>0) break;
            int i=k+1,j=n-1;
            while(i<j){
                if(nums[i]+nums[j]+t==0){
                    ans.push_back({t,nums[i],nums[j]});
                    while(++i<j && nums[i]==nums[i-1]);
                    while(i<--j && nums[j]==nums[j+1]);
                }else if(nums[i]+nums[j]+t<0)
                    i++;
                else
                    j--;
            }
            while(k<n-2 && nums[k]==t) k++;
        }
        return ans;
    }
};