给你一个长度为 n 的整数数组 nums 和 一个目标值 target。 请你从 nums 中选出三个整数,使它们的和与 target 最接近。 返回这三个数的和。
假定每组输入只存在恰好一个解。
示例 1: 输入:nums = [-1,2,1,-4], target = 1 输出:2 解释:与 target 最接近的和是 2 (-1 + 2 + 1 = 2)。
示例 2: 输入:nums = [0,0,0], target = 1 输出:0 解释:与 target 最接近的和是 0(0 + 0 + 0 = 0)。 提示: 3 <= nums.length <= 1000 -1000 <= nums[i] <= 1000 -104 <= target <= 104
|
解题思路
这一题和上一题非常相似,只是单纯将0变为一个指定值target,那么可以沿用上一题的思路,不断在比较中确定一个最接近的值即可。
解决步骤依旧是先排序,然后选择一个值nums[i],选择其右边的值为Left,选择nums的最后一个值为Right,即Left = i + 1; Right = nums.length - 1
。
比较三数之和,如果比target大,那么Right--
,如果比target小,那么Left++
,使用target与sum差值的绝对值来更新结果。
代码
class Solution { public int threeSumClosest(int[] nums, int target) { Arrays.sort(nums); int n = nums.length; int best = 10000000;
for (int i = 0; i < n; ++i) { if (i > 0 && nums[i] == nums[i - 1]) { continue; } int j = i + 1, k = n - 1; while (j < k) { int sum = nums[i] + nums[j] + nums[k]; if (sum == target) { return target; } if (Math.abs(sum - target) < Math.abs(best - target)) { best = sum; } if (sum > target) { int k0 = k - 1; while (j < k0 && nums[k0] == nums[k]) { --k0; } k = k0; } else { int j0 = j + 1; while (j0 < k && nums[j0] == nums[j]) { ++j0; } j = j0; } } } return best; } }
|
当然这个解法仍旧有两个可以优化的点
设s = nums[i] + nums[i+1] + nums[i+2]
。如果s > target
,由于数组已经排序,后面无论怎么选,选出的三个数的和不会比s还小,所以不会找到比s更优的答案了。
所以只要s>target
,就可以直接break外层循环了。在break前判断s是否离target更近,如果更近,那么更新best为s。
设s = nums[i] + nums[n-2] + nums[n-1]
。如果s < target
,由于数组已经排序,nums[i]加上后面任意两个数都不超过s,所以下面的双指针就不需要跑了,无法找到比s更优的答案。
但是后面还有更大的nums[i],可能找到一个离target更近的三数之和,所以还需要继续枚举,continue外层循环。在continue前判断s是否离target更近,如果更近,那么更新best为s。