给定一个包括 n 个整数的数组 nums 和 一个目标值 target。找出 nums 中的三个整数,使得它们的和与 target 最接近。返回这三个数的和。假定每组输入只存在唯一答案。
例如,给定数组 nums = [-1,2,1,-4], 和 target = 1.
与 target 最接近的三个数的和为 2. (-1 + 2 + 1 = 2).
解题思路:
- 本题目因为要计算三个数,如果靠暴力枚举的话时间复杂度会到O(n^3),需要降低时间复杂度;
- 首先进行数组排序,时间复杂度O(nlogn);
- 在数组nums中,进行遍历,每遍历一个值利用其下标i,形成一个固定值nums[i];
- 再使用前指针指向start = i + 1处,后指针指向end = nums.length - 1处,也就是结尾处;
- 根据 sum = nums[i] + nums[start] + nums[end] 的结果,判断sum与目标target的距离,如果更近则更新结果ans;
- 同时判断sum与target的大小关系,因为数组有序,如果sum > target 则 end–,如果sum < target 则start++,如果sum == target 则说明距离为0直接返回结果;
- 整个遍历过程,固定值为n次,双指针为n次,时间复杂度为O(n^2);
- 总时间复杂度:O(nlogn) + O(n^2) = O(n^2)。
代码实现:
import java.util.Arrays;
/**
* @author zhangyan
* @date 2018/10/8 8:16
*/
public class ThreeSumClosest {
public int threeSumClosest(int[] nums, int target) {
Arrays.sort(nums);
int ans = nums[0] + nums[1] + nums[2];
for(int i=0;i<nums.length;i++) {
int start = i+1, end = nums.length - 1;
while(start < end) {
int sum = nums[start] + nums[end] + nums[i];
if(Math.abs(target - sum) < Math.abs(target - ans))
ans = sum;
if(sum > target)
end--;
else if(sum < target)
start++;
else
return ans;
}
}
return ans;
}
// 测试
public static void main(String[] args) {
int[] num = new int[]{-1,2,1,-4};
System.out.println(threeSumClosest(num,1));
}
}
有问题欢迎留言哦
参考:
https://leetcode-cn.com/problems/3sum-closest/solution/
https://draw.mdnice.com/algorithm/16.html