题目:
给定一个包括 n 个整数的数组 nums 和 一个目标值 target。找出 nums 中的三个整数,使得它们的和与 target 最接近。返回这三个数的和。假定每组输入只存在唯一答案。
例如,给定数组 nums = [-1,2,1,-4], 和 target = 1.
与 target 最接近的三个数的和为 2. (-1 + 2 + 1 = 2).
python代码:
class Solution:
def threeSumClosest(self, nums: List[int], target: int) -> int:
nums.sort()
ans = nums[0] + nums[1] + nums[2]
for i in range(len(nums)-1):
start,end = i+1,len(nums)-1
while(start < end):
summ = nums[start] + nums[end] + nums[i]
if(abs(target - summ) < abs(target - ans)):
ans = summ
if(summ > target):
end-=1
elif(summ < target):
start+=1
else:
return ans
return ans
思路:
本题目因为要计算三个数,如果靠暴力枚举的话时间复杂度会到
O
(
n
3
)
O(n^3)
O(n3),需要降低时间复杂度
首先进行数组排序,时间复杂度
O
(
n
l
o
g
n
)
O(nlogn)
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(n^2)
O(n2)
总时间复杂度:
O
(
n
l
o
g
n
)
O(nlogn)
O(nlogn) +
O
(
n
2
)
O(n^2)
O(n2)=
O
(
n
2
)
O(n^2)
O(n2)