三数之和

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

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

示例:

给定数组 nums = [-1, 0, 1, 2, -1, -4],

满足要求的三元组集合为:
[
  [-1, 0, 1],
  [-1, -1, 2]
]

思路:

排序 + 双指针
本题的难点在于如何去除重复解。

算法流程:

  1. 对于数组长度 n,如果数组为 null 或者数组长度小于 3,返回 []。
  2. 对数组进行排序。
  3. 遍历排序后数组:
    若 nums[i]>0:因为已经排序好,所以后面不可能有三个数加和等于 0,直接返回结果。
  4. 对于重复元素:跳过,避免出现重复解

令左指针 L=i+1,右指针 R=n-1,当 L<R时,执行循环:

  • 当 nums[i]+nums[L]+nums[R]==0,执行循环,判断左界和右界是否和下一位置重复,去除重复解。并同时将 L,R 移到下一位置,寻找新的解
  • 若和大于 0,说明 nums[R] 太大,R 左移,若和小于 0,说明 nums[L] 太小,L 右移
  • 复杂度分析
  1. 时间复杂度:数组排序O(nlogn),遍历数组O(n),双指针遍历O(n)。总=O(nlogn)+O(n)*O(n)=O(n2)

  2. 空间复杂度:O(1)
Class Solution{

public:
    vector<vector<int>> ThreeSum(vector<int>& nums)
    {
        sort(nums.begin(),nums.end());//排序
        vector<vector<int>> res;
        int nCount=nums.size();
        if(nCount < 3)
            return res;
        int i,j,k;
        for(k = 0; k < nums.size(); k++)
        {
            if(nums[k] > 0)//#1.because of j>i>k.
                break;
            if(k > 0 && nums[k] == nums[k+1]);//#2.skip the same nums[k]
                continue;

            i=k+1;
            j=nCount-1;
            while(i<j)
            {
                int sum=nums[i]+nums[j]+nums[k];
                if(sum > 0)
                    --j;
                else if(sum<0)
                    ++i;
                else
                {
                    res.push_back({nums[i],nums[j],nums[k]});
                    while(i < j && nums[i] == nums[i+1])
                          i++;
                    while(i < j && nums[j] == nums[j-1])
                          --j;

                    ++i;
                    --j;
                }
            }
        }
        return res;        
    }      
}

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值