题目大意
给定一个数组与target,然后从数组中选三个数使它们的和与target最接近。
分析
其实思路与第十五题比较相似,只是在双指针移动的时候要用三数之和与target做差来决定指针移动,比如大于零,那么说明三数之和比较大,需要将right向左移动。
注意,首先要排序,然后可以剪枝。比如第五行,我们需要让nums[i]与左边的nums[i-1]相等,才能对这种情况剪枝,不能通过nums[i]与右边的nums[i+1]相等来剪枝,比如[1,1,1],当最左边的那个1为起点搜索过之后的情况,才可以剪枝,如果我们判断它与右边相等而剪枝的话,那么以1为起点的搜索空间中将不存在1,1,1(3个1)的情况。
代码
int threeSumClosest(vector<int>& nums, int target) {
int ans = 0, tmp = 0x3f3f3f3f;
sort(nums.begin(), nums.end());
for(int i = 0; i < nums.size() - 2; i++) {
if(i != 0 && nums[i] == nums[i - 1]) {
continue;
}
int left = i + 1, right = nums.size() - 1;
while(left < right) {
int sum = nums[i] + nums[left] + nums[right];
int tag = sum - target;
if(abs(tag) < tmp) {
ans = sum;
tmp = abs(sum - target);
}
if(tag > 0) {
right--;
while(left < right && nums[right] == nums[right + 1]) {
right--;
}
continue;
} else {
left++;
while(left < right && nums[left] == nums[left - 1]) {
left++;
}
continue;
}
}
}
return ans;
}