leetcode 最接近的三数之和

最新推荐文章于 2024-11-11 20:59:17 发布

weixin_45776003

最新推荐文章于 2024-11-11 20:59:17 发布

阅读量78

点赞数

文章标签： leetcode 数据结构算法

本文链接：https://blog.csdn.net/weixin_45776003/article/details/104682785

版权

给定一个包括 n 个整数的数组 nums 和一个目标值 target。找出 nums 中的三个整数，使得它们的和与 target 最接近。返回这三个数的和。假定每组输入只存在唯一答案。

例如，给定数组 nums = [-1，2，1，-4], 和 target = 1.
与 target 最接近的三个数的和为 2.
(-1 + 2 + 1 = 2).

本题可以用暴力法，即三次循环穷举找到答案，时间复杂度为O(n^3)
故需要对其进行优化
与上一题：三数之和类似，可以采用三指针法。上篇链接
将原数组排序后，第一个指针i遍历数组，另外两个指针j和k位于i右面区域的两边，根据三数和的大小右移或左移，由两个变量more和less存放和与target的差值，more存放比target大的最小差值，less存放比target小的最小差值，最后由较小者决定返回的答案
代码如下:

class Solution {
public:
    int threeSumClosest(vector<int>& nums, int target) 
    {
        int ans=target;
        int l=size(nums);
        int less=INT_MAX;
        int more=INT_MAX;
        if(l<3)
            return 0;
        sort(nums.begin(),nums.end());
        for(int i=0;i<l;i++)
        {
            int j=i+1,k=l-1;
            while(j<k)
            {
                int sum=nums[i]+nums[j]+nums[k];
                
                if(sum==target)
                    return target;
                if(sum<target)
                {
                    if(target-sum<less)
                    {
                        less=target-sum;
                    }
                    j++;
                }
                if(sum>target)
                {
                    if(sum-target<more)
                    {
                        more=sum-target;
                    }
                    k--;
                }
            }         
        }
        ans=more>less?target-less:target+more;
        return ans;
    }
};