-
Notifications
You must be signed in to change notification settings - Fork 614
/
16.py
44 lines (33 loc) · 1.08 KB
/
16.py
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
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
'''
Given an array nums of n integers and an integer target, find three integers in nums such that the sum is closest to target. Return the sum of the three integers. You may assume that each input would have exactly one solution.
Example:
Given array nums = [-1, 2, 1, -4], and target = 1.
The sum that is closest to the target is 2. (-1 + 2 + 1 = 2).
'''
class Solution(object):
def threeSumClosest(self, nums, target):
"""
:type nums: List[int]
:type target: int
:rtype: int
"""
nums.sort()
result, min_diff = 0, float('inf')
for index in range(len(nums)-1):
left = index + 1
right = len(nums) - 1
while left < right:
currSum = nums[index] + nums[left] + nums[right]
diff = abs(target - currSum)
if diff == 0:
return target
if diff < min_diff:
min_diff = diff
result = currSum
if currSum < target:
left += 1
else:
right -= 1
return result
# Space: O(1)
# Time: O(N^2)