LeetCode:三数之和/最接近的三数之和

题目:
给定一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]

思路:
这道题用暴力法三层嵌套for循环的时间复杂度太高,不可取。
应该思考这三个数满足的特征,以此来限制条件减少循环次数。

以0为分界,三个数中全为0或有正有负才行。
由此就要考虑,先排序成有序数组。选定一个数为c位,用双指针前后寻找剩下满足条件的两个数。

1.什么情况下不存在满足条件数组
元素个数<3; 排序后的尾部元素<0; 排序后的首部元素>0 。
return
2. 选哪个数据为中值参照?
以左边元素为参照,在它后面的序列中选择符合条件的左右值。
(1) 参照元素>0时,后面全大于0, 查找结束.break
(2)参照元素重复时,应跳过该元素,continue
(3) 定义好左右指针
求sum
sum =0 ,push_back({}),l++,r–
sum <0 ,l++
sum > 0,r–
碰到相邻的重复元素,也应跳过。
3.查找结束,返回存储容器。

    vector<vector<int>> threeSum(vector<int>& nums) {
        int len=nums.size();
        vector<vector<int>>res;
        res.clear();
        sort(nums.begin(),nums.end());
        if(len<3||nums[0]>0||nums[len-1]<0)
            return res;
        for(int i=0;i<len-2;i++)
        {
            if(nums[i]>0)
                break;
            if(i>0&&(nums[i]==nums[i-1]))continue;            
            int left = i + 1;
            int right = len - 1;
            while(left < right )
            {
                int sum = nums[i] + nums[left]+nums[right];
                if(sum == 0)
                {
                   // vector<int> tmp;
                   // tmp.push_back(nums[i]);
                   // tmp.push_back(nums[left]);
                   // tmp.push_back(nums[right]);
                   // res.push_back(tmp);
                    res.push_back({nums[i],nums[left],nums[right]});
                    left++;
                    right--;
                    while(left<right&&(nums[left]==nums[left-1]))left++;
                    while(left<right&&(nums[right]==nums[right+1]))right--;
                }
                else if(sum<0)
                {
                    left++;
                     while(left<right&&(nums[left]==nums[left-1]))left++;
                }
                else
                {
                    right--;
                    while(left<right&&(nums[right]==nums[right+1]))right--;
                }               
            }
        }
        return res;
    }

类似的题有最接近的三数之和 :
给定一个包括 n 个整数的数组 nums 和 一个目标值 target。找出 nums 中的三个整数,使得它们的和与 target 最接近。返回这三个数的和。假定每组输入只存在唯一答案。
例如,给定数组 nums = [-1,2,1,-4], 和 target = 1.
与 target 最接近的三个数的和为 2. (-1 + 2 + 1 = 2).

还是卡边界处理了特定情况后处理一般情况
用一个Min值来控制是否记录最接近的值。

    int threeSumClosest(vector<int>& nums, int target) {
        sort(nums.begin(),nums.end());
        int res=0,min = INT_MAX;
        int len = nums.size();
        if(len<=3)
        {
            for(int i=0;i<len;i++)
                res+=nums[i];
            return res;
        }
        if((nums[0]+nums[1]+nums[2])>target)
            return nums[0]+nums[1]+nums[2];
        else if((nums[len-1]+nums[len-2]+nums[len-3])<target)
            return nums[len-1]+nums[len-2]+nums[len-3]; 
        else
        {
            for(int i=0;i<len-2;i++)
            {
                int left=i+1;
                int right = len-1;
                if((nums[i]+nums[i+1]+nums[i+2])>target)
                {
                     if((nums[i]+nums[i+1]+nums[i+2]-target)<min)
                        res = nums[i]+nums[i+1]+nums[i+2];
                     break;
                }                    
                else if( (nums[i]+nums[len-1]+nums[len-2]) <target)
                {
                     if( target-(nums[i]+nums[len-1]+nums[len-2]) < min)
                     {
                         res = nums[i]+nums[len-1]+nums[len-2];
                         min = target-(nums[i]+nums[len-1]+nums[len-2]);
                     }
                }                    
                else
                {
                    while(left<right)
                    {
                         int sum = nums[i]+nums[left]+nums[right];
                         if(sum == target)
                            return target;
                         else if(sum < target)
                         {
                             left++;
                         }
                        else
                        {
                            right--;
                        }
                        if(abs(sum-target)<min)                         
                        {
                            res = sum;
                            min = abs(sum-target);
                        }
                    }                 
                }
            }
        }
        return res;   
    }
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值