[Leetcode]_16 3Sum Closest

最新推荐文章于 2017-09-03 22:46:37 发布

ltree98

最新推荐文章于 2017-09-03 22:46:37 发布

阅读量716

点赞数

分类专栏： Leetcode LeetCode 刷刷刷文章标签： leetcode

本文链接：https://blog.csdn.net/lttree/article/details/70217598

版权

Leetcode 同时被 2 个专栏收录

59 篇文章 0 订阅

订阅专栏

LeetCode 刷刷刷

47 篇文章 0 订阅

订阅专栏

/**
 *  Index: 16
 *  Title: 3Sum Closest
 *  Author: ltree98
 **/

第15题的变种，
之前是三个数相加等于0，
这次让三个数相加等于给定的数，若相等，直接返回给定的数。
否则就一直算。

注意：
此处ans初始值设置为INT_MAX,
但是ans类型不能是INT，
因为当target是负数时，INT_MAX-target 会溢出。

class Solution {
public:
    int threeSumClosest(vector<int>& nums, int target) {
        std::sort(nums.begin(), nums.end());
        long ans = INT_MAX;

        for(int i = 0; i < nums.size(); )   {
            int l = i+1, h = nums.size()-1;
            while(l < h)    {
                int temp = nums[i] + nums[l] + nums[h];
                if(temp == target)  {
                    return target;
                }
                else if(temp < target)   {
                    if(abs(ans-target) > abs(temp-target))
                        ans = temp;
                    l++;
                }
                else {
                    if(abs(ans-target) > abs(temp-target))
                        ans = temp;
                    h--;
                }
            }

            i++;
            while(i < nums.size() && nums[i] == nums[i-1])
                i++;
        }

        return ans;
    }
};