LeetCode - 16. 3Sum Closest

Posted 2020-06-14

16. 3Sum Closest 

Problem‘s Link

 ----------------------------------------------------------------------------

Mean: 

给定一个整数序列和一个目标数，在序列中找出三个数，使得这三个数的和与目标数的差最小.

analyse:

O(n^2)的做法.

如果你找到了更好的做法，请赐教^_^

Time complexity: O(N)

 

view code

/**
* -----------------------------------------------------------------
* Copyright (c) 2016 crazyacking.All rights reserved.
* -----------------------------------------------------------------
*       Author: crazyacking
*       Date  : 2016-02-17-08.50
*/
#include <queue>
#include <cstdio>
#include <set>
#include <string>
#include <stack>
#include <cmath>
#include <climits>
#include <map>
#include <cstdlib>
#include <iostream>
#include <vector>
#include <algorithm>
#include <cstring>
using namespace std;
typedef long long(LL);
typedef unsigned long long(ULL);
const double eps(1e-8);

class Solution
{
public:
   int threeSumClosest(vector<int>& nums, int target)
   {
       int si=nums.size();
       if(si<3) return 0;
       if(si==3) return (nums[0]+nums[1]+nums[2]);
       int64_t res=INT_MAX,low,high,sum,tsum;
       sort(nums.begin(),nums.end());

       for(int i=0;i<si;++i)
       {
           low=i+1;
           high=si-1;
           tsum=target-nums[i];
           while(low<high)
           {
               sum=(nums[i]+nums[low]+nums[high]);
               if(abs(target-res)>abs(target-sum))
                   res=sum;
                   
               if(nums[low]+nums[high]<tsum)
                   ++low;
               else if(nums[low]+nums[high]>tsum)
                   --high;
               else
                   return target;
           }
       }
       return (int)res;
   }
};

int main()
{
   Solution solution;
   int n,target;
   while(cin>>n>>target)
   {
       vector<int> ve;
       for(int i=0;i<n;++i)
       {
           int tmp;
           cin>>tmp;
           ve.push_back(tmp);
       }
       int ans=solution.threeSumClosest(ve,target);
       cout<<ans<<endl;
   }
   return 0;
}