DailyChallenge
16. 最接近的三数之和
给定一个包括 n 个整数的数组 nums 和 一个目标值 target。找出 nums 中的三个整数,使得它们的和与 target 最接近。返回这三个数的和。假定每组输入只存在唯一答案。
示例:
输入:nums = [-1,2,1,-4], target = 1
输出:2
解释:与 target 最接近的和是 2 (-1 + 2 + 1 = 2) 。
提示:
- 3 <= nums.length <= 10^3
- -10^3 <= nums[i] <= 10^3
- -10^4 <= target <= 10^4
链接:
Solution
题目要求找到与目标值 target 最接近的三元组,这里的「最接近」即为差值的绝对值最小。我们可以考虑直接使用三重循环枚举三元组,找出与目标值最接近的作为答案,时间复杂度为 O(N^3)。然而本题的 N 最大为 1000,会超出时间限制。
那么如何进行优化呢?我们首先考虑枚举第一个元素 a,对于剩下的两个元素 b 和 c,我们希望它们的和最接近target−a。对于 b 和 c,如果它们在原数组中枚举的范围(既包括下标的范围,也包括元素值的范围)没有任何规律可言,那么我们还是只能使用两重循环来枚举所有的可能情况。因此,我们可以考虑对整个数组进行升序排序,这样一来:
- 假设数组的长度为 n,我们先枚举 a,它在数组中的位置为 i;
- 为了防止重复枚举,我们在位置 [i+1,n) 的范围内枚举 b 和 c。
借助双指针,我们就可以对枚举的过程进行优化。我们用 left 和 right 分别表示指向 b 和 c 的指针,初始时,left 指向位置 i+1,即左边界;right 指向位置 n−1,即右边界。在每一步枚举的过程中,我们用 a+b+c 来更新答案,并且:
- 如果 a+b+c > target,那么就将 right 左移
- 如果 a+b+c < target,那么就将 left 右移
- 如果 a+b+c = target,返回target
class Solution {
public int threeSumClosest(int[] nums, int target) {
// 不能去Interger.MAX_VALUE
int min = 100000000;
int sum = 0;
// 排序
Arrays.sort(nums);
for(int i = 0; i < nums.length - 2; i++){
int left = i + 1, right = nums.length - 1;
if (i > 0 && nums[i] == nums[i - 1]) {
continue;
}
while(left < right){
sum = nums[i] + nums[left] + nums[right];
if(Math.abs(sum - target) < Math.abs(min - target)){
min = sum;
}
if(sum == target){
return sum;
}else if(sum < target){
int temp = left + 1;
while( temp < right && nums[temp] == nums[left]){
temp++;
}
left = temp;
}else if(sum > target){
int temp = right - 1;
while( temp > left && nums[temp] == nums[right]){
temp--;
}
right = temp;
}
}
}
return min;
}
}