[LeetCode]3Sum Closest

来源:互联网 发布:繁体转简体软件 编辑:程序博客网 时间:2024/06/14 09:21

题目要求如下:

Given an array S of n integers, find three integers in S such that the sum is closest to a given number, target. Return the sum of the three integers. You may assume that each input would have exactly one solution.

For example, given array S = {-1 2 1 -4}, and target = 1.    The sum that is closest to the target is 2. (-1 + 2 + 1 = 2).

 

这道题的解法和3Sum这道题很相像,这里就不说得太细了,解题思路如下:

因为要求3个数的和最接近target,所以我们先固定其中1个数,再用求“和为某值的2个数的组合”的解法,就能把剩下的2个数求出

来。此,先对数组进行非递减排序,这样整个数组的数就由小到大排列。i 的取值由 0 至 n-1,对每一个i,我们求当num[i]是解当

中的其中一个数时,其他的2个数。设有指针p指向数组头(实际只要p从i+1开始),q指向数组尾,cur = num[i] + num[p]+ num[q],那

么当前值与所求值target之间的差距就为temp = cur - target,如果abs(temp) < min,更新min = abs(temp)和当前解res = cur。

现在分为3种情况:

1. temp == 0,说明与target最接近,因为题目已经说明只有一个解,所以直接返回cur;

2. temp < 0,说明num[p]太小了,将p向后移动一个位置;

3. temp > 0,说明num[q]太大了,将q向前移动一个位置。

如此循环直到p == q为止。

 

在求解过程中,需要注意一些细节:

1. 如果数组元素个数num.size() == 3,直接返回这3个数的和,因为只有这个解;

2. 如果i != 0,而且num[i] == num[i - 1],说明刚才求过的数和现在要求的一样,没必要再重复一次,直接跳过;

3. p可以直接从i + 1位置开始,因为在i 之前的位置j 上的数如果也是解的一部分,那么在求j的时候,肯定把i位置上的数也算过了;

4. 在移动p和q的时候要注意不能等于i。

 

以下是我的代码,欢迎各位大牛指导交流~

AC,Runtime: 44 ms

 

//LeetCode_3Sum Closet//Written by zhou//2013.11.25class Solution {public:    void swap(int &a, int &b)    {        int c = a;        a = b;        b = c;    }    void qsort(vector<int> &num, int left, int right)    {        int i = left - 1, j = right;        int pivot = num[right];        while(i < j)        {            while(++i <= right && num[i] < pivot)            {                //do nothing            }            while(--j >= left && num[j] > pivot)            {                //do nothing            }            if (i < j)            {                swap(num[i],num[j]);            }        }        swap(num[i],num[right]);                if (left < i - 1)            qsort(num,left,i-1);        if (i + 1 < right)            qsort(num,i+1,right);    }    int threeSumClosest(vector<int> &num, int target) {        // IMPORTANT: Please reset any member data you declared, as        // the same Solution instance will be reused for each test case.        if (num.size() == 3)return num[0] + num[1] + num[2];        //非递减(递增)排序        qsort(num,0,num.size()-1);                int min = 0x7fffffff;        int res = 0;        int cur = 0;                for (int i = 0; i < num.size(); ++i)        {//刚才找过了同值的,直接跳过            if (i != 0 && num[i] == num[i-1])                continue;                        int p = i + 1, q = num.size() - 1;                        while(p < q)            {                cur = num[i] + num[p] + num[q];int temp = cur - target;if (temp == 0)        //刚好相等直接返回return cur;                if (abs(temp) < min)   //更新值                {                    min = abs(temp);                    res = cur;                }//移动指针进行下一步操作if (temp < 0){while (++p < q && p == i){//do nothing}}else  //temp > 0{while (--q > p && q == i){//do nothing}}            }        }                return res;    }};