您的位置:首页 > 其它

[LeetCode]3Sum Closest

2013-11-25 12:39 399 查看
题目要求如下:

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.25

class 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;
}
};
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: