题目大意
3sum问题的变种,寻找与目标数字最近的那一组数,返回三数之和
解题思路
一样的遍历每个数,对剩余数组进行双指针扫描。区别仅仅在于当:
sum = A[left] + A[right]
(1) sum = target时直接返回
(2) sum != target时,在相应移动left/right指针之前,先计算abs(sum-target)的值,并更新结果。
时间复杂度:O(n log n)(排序)+ O(n^2)= O(n^2)
代码
class Solution:
def threeSumClosest(self, nums, target):
""" :type nums: List[int] :type target: int :rtype: int """
nums.sort() # 先排序
closest_sum = sys.maxsize
for i in range(len(nums)-2): # 遍历至倒数第三个,后面两个指针
if i == 0 or nums[i] > nums[i-1]: # 排除相同数和刚开始的时候
left = i + 1
right = len(nums) - 1
while left < right:
diff = nums[left] + nums[right] + nums[i] - target
if abs(diff) < abs(closest_sum):
closest_sum = diff
if diff == 0:
return target
elif diff < 0:
left += 1
else:
right -= 1
return closest_sum + target # 原三数之和
总结
该题显然是考验双指针,双指针思路清晰,更容易理解。