LeetCode16:3Sum Closest

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

#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;

//要求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为止。
class Solution {
public:
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.

//如果数组元素个数num.size() == 3，直接返回这3个数的和，因为只有这个解；
if (num.size() == 3)
return num[0] + num[1] + num[2];

//非递减（递增）排序
sort(num.begin(), num.end());

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;
// p可以直接从i + 1位置开始，因为在i 之前的位置j 上的数如果也是解的一部分，那么在求j的时候，肯定把i位置上的数也算过了；
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;
}

//移动指针进行下一步操作 ,注意在移动p和q的时候，不能等于i
if (temp < 0)
{
while (++p < q && p == i)
{
//do nothing
}
}
else  //temp > 0
{
while (--q > p && q == i)
{
//do nothing
}
}
}
}

return res;
}
};

int main()
{
vector<int> num1 = { -1, 2, 1,-4 };
Solution sol;
int t = sol.threeSumClosest(num1, 1);
cout << t << endl;
system("pause");
return 0;
}