【Leetcode】【python】3Sum Closest

题目大意

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)

代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
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 # 原三数之和

总结

该题显然是考验双指针,双指针思路清晰,更容易理解。