一、题目描述
给定一个包括 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
二、解题思路
暴力法的时间复杂度是 O(n^3),可以先固定一个值,然后寻找后两个值时可采取双指针的方法,将总的时间复杂度优化到 O(n^2)。
首先我们先对原数组进行排序,初始化最接近的和为前三个元素之和,
closestNum = nums[0] + nums[1] + nums[2]
接下来开始遍历,第一个元素的下标i从0到 nums.length-2,然后利用双指针,left 指向 i + 1, right 指向 nums.length - 1。
当三数之和与target更接近时,更新closestNum;
当三数之和大于target时,right-1;
当三数之和小于target时,left-1;
当三数之和等于target时,直接返回target;
三、代码
import java.util.Arrays;
public class Solution {
public int threeSumClosest(int[] nums, int target) {
Arrays.sort(nums);
int closestNum = nums[0] + nums[1] + nums[2];
for(int i=0;i<nums.length-2;i++){
int left = i + 1, right = nums.length - 1;
while(left<right){
int threeSum = nums[left] + nums[right] + nums[i];
if(Math.abs(threeSum-target)<Math.abs(closestNum-target)){
closestNum=threeSum;
}
if(threeSum>target){
right--;
} else if(threeSum<target){
left++;
} else {
return threeSum;
}
}
}
return closestNum;
}
}
四、复杂度分析
时间复杂度: O(n^2)
空间复杂度:
我们忽略存储答案的空间,额外的排序的空间复杂度为 O(logN)。
然而我们修改了输入的数组 nums,在实际情况下不一定允许,因此也可以看成使用了一个额外的数组存储了nums 的副本并进行排序,空间复杂度为 O(N)。
欢迎关注微信公众号【算法攻城师】