Leedcode 15三数之和

给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有和为 0 且不重复的三元组。

注意:答案中不可以包含重复的三元组。

示例 1:

输入:nums = [-1,0,1,2,-1,-4]
输出:[[-1,-1,2],[-1,0,1]]
示例 2:

输入:nums = []
输出:[]
示例 3:

输入:nums = [0]
输出:[]
排序 + 双指针

1、枚举每个数,表示该数nums[i]已被确定,在排序后的情况下,通过双指针l,r分别从左边l = i + 1和右边n - 1往中间靠拢,找到nums[i] + nums[l] + nums[r] == 0的所有符合条件的搭配
2、在找符合条件搭配的过程中,假设sum = nums[i] + nums[l] + nums[r]
若sum > 0,则r往左走,使sum变小
若sum < 0,则l往右走,使sum变大
若sum == 0,则表示找到了与nums[i]搭配的组合nums[l]和nums[r],存到ans中
3、判重处理
确定好nums[i]时,l 需要从i + 1开始
当nums[i] == nums[i - 1],表示当前确定好的数与上一个一样,需要直接continue
当找符合条件搭配时,即sum == 0,需要对相同的nums[l]和nums[r]进行判重出来
时间复杂度 O(n2)


class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        sort(nums.begin(),nums.end());
        vector<vector<int>> res;
        for(int i=0;i<nums.size();i++){
            if(i&&nums[i]==nums[i-1])continue;
            for(int j=i+1,k=nums.size()-1;j<k;j++){
                if(j>i+1&&nums[j]==nums[j-1])continue;
                while(j<k-1&&nums[i]+nums[j]+nums[k-1]>=0)k--;
                if(nums[i]+nums[j]+nums[k]==0){
                    res.push_back({nums[i],nums[j],nums[k]});
                }

            }
        }
       return res;
    }
      
};

版权声明:本文为qq_43738331原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。