16.3Sum Closest

来源:互联网 发布:python 替换分隔符 编辑:程序博客网 时间:2024/06/05 05:06

给定一个整型数组和一个目标值,找出数组中‘累加和’与‘目标值’最近的三个数的和,假设只存在唯一解

-例子:

Input:  array S={-1,2,1,-4},target = 1Output: 2

刚看到这一题我的想法是3层循环遍历,设置一个变量始终保存 与目标值差绝对值最小的数字,时间复杂度为O(n3),需要考虑降低时间复杂度。**
以前LeetCode上 3Sum的题目,对于求和数组可以先排序,得到一个有序数组。

解题思路

假设数组S有N的元素。

  • 首先选定三个元素:i,m = i+1,n = N-1,其中i为基准值从0开始,并设置变量temp保存最接近的数字。
  • 求出S[i] + S[m] + S[n],并将其保存至变量tmp。

    if(tmp > target){     n --;}else{     m ++;}

    每次移动后,比较abs(S[i] + S[m] + S[n] - target) 与abs(tmp - target)的大小,进行替换;当m与n交叉时,结束本轮循环,如果中间遇到S[i] + S[m] + S[n] - target = 0,直接返回target。

  • 然后i++,开启第二轮循环,直至i = N - 2结束,返回tmp。

实现:

int threeSumClosest(vector<int>& nums, int target) {        int tmp = nums[0] + nums[1] + nums[nums.size() -1];        sort(nums.begin(),nums.end());        for(int i = 0;i < nums.size() - 2; i++)        {            int m = i + 1;            int n = nums.size() -1;            while(m < n)            {                int curTmp = nums[i] + nums[m] + nums[n];                if(abs(curTmp - target) < abs(tmp - target))                {                    tmp = curTmp;                }                if(curTmp > target)                {                    n --;                }                else if(curTmp < target)                {                    m ++;                }                else                {                    return target;                }            }        }        return tmp;    }
原创粉丝点击