2021-6-28 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]
输出:[]

提示:
0 <= nums.length <= 3000
-105 <= nums[i] <= 105

题解:

「不重复」的本质是什么?我们保持三重循环的大框架不变,只需要保证:

  1. 第二重循环枚举到的元素不小于当前第一重循环枚举到的元素;
  2. 第三重循环枚举到的元素不小于当前第二重循环枚举到的元素;
  3. 同时,对于每一重循环而言,相邻两次枚举的元素不能相同。
    也就是说,我们枚举的三元组 (a,b,c) 满足a≤b≤c,保证了只有(a,b,c) 这个顺序会被枚举到,而(b,a,c)、(c,b,a) 等等这些不会,这样就减少了重复。要实现这一点,我们可以将数组中的元素从小到大进行排序。

双指针法
思路: 固定 3 个指针中最左(最小)数字的指针 first,双指针 second,third 分设在数组索引 (first, len(nums))两端,通过双指针交替向中间移动,记录对于每个固定指针 first 的所有满足 nums[first] + nums[second] + nums[third] = 0 的 i,j 组合:

  1. 当 first> 0且nums[first] = nums[first - 1]时即跳过此元素nums[first]:因为已经将 nums[first - 1]的所有组合加入到结果中,本次双指针搜索只会得到重复组合。
  2. second,third 分设在数组索引 (first, len(nums)) 两端,为了防止重复,当 second> first+1且nums[second] = nums[second - 1]时即跳过此元素nums[first]。
  3. 令target = -nums[first] ,当second < third 时循环计算nums[second] + nums[third]的值,并按照以下规则执行双指针移动:
    1. 当nums[second] + nums[third] <target时,break;
    2. 当nums[second] + nums[third] > target时,third–;
    3. 当nums[second] + nums[third] = target时,记录组合[first, second, third]至res,并break;

复杂度分析:
时间复杂度 O(N^2):其中固定指针k循环复杂度O(N),双指针 i,j 复杂度 O(N)。
空间复杂度 O(1):指针使用常数大小的额外空间。

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        sort(nums.begin(),nums.end());
        vector<vector<int>> result;
        int left=0;
        int right=nums.size()-1;
        for(int first=0;first<nums.size();first++){
            if(first>0&&nums[first]==nums[first-1]){
                continue;
            }
            for(int second=first+1;second<nums.size();second++){
                if(second>first+1&&nums[second]==nums[second-1]){
                    continue; 
                }
                int third=nums.size()-1;
                int target=-nums[first];
                while(second<third){
                    if(nums[third]+nums[second]<target){
                        break;
                    }
                    else if(nums[third]+nums[second]>target){
                        third--;
                    }
                    else{
                        result.push_back({nums[first],nums[second],nums[third]});
                        break;
                    }
                }
            }
        }
        return result;
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值