题目描述
给定一个包括 n 个整数的数组 nums 和 一个目标值 target。找出 nums 中的三个整数,使得它们的和与 target 最接近。返回这三个数的和。假定每组输入只存在唯一答案。
例如,给定数组 nums = [-1,2,1,-4], 和 target = 1.
与 target 最接近的三个数的和为 2. (-1 + 2 + 1 = 2).
思路
设置一个返回结果,用于更新最接近的目标的三个数之和
同样的操作,固定一个数,剩下来的二分查找
- 首先进行数组排序,时间复杂度 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)
class Solution:
def threeSumClosest(self, nums, target):
if not nums or len(nums) < 3:
return None
nums.sort()
ans = nums[0] + nums[1] + nums[2]
for i in range(len(nums) - 2):
left, right = i + 1, len(nums) - 1
while left < right:
sums = nums[i] + nums[left] + nums[right]
if abs(target - ans) > abs(target - sums):
ans = sums
if sums > target:
right -= 1
elif sums < target:
left += 1
else:
return ans
return ans
AC16
网友评论