题目描述:
给定一个包括 n 个整数的数组 nums 和 一个目标值 target。找出 nums 中的三个整数,使得它们的和与 target 最接近。返回这三个数的和。假定每组输入只存在唯一答案。
示例:
输入:nums = [-1,2,1,-4], target = 1
输出:2
解释:与 target 最接近的和是 2 (-1 + 2 + 1 = 2) 。
提示:
3 <= nums.length <= 10^3
-10^3 <= nums[i] <= 10^3
-10^4 <= target <= 10^4
解题思路一
超时解法。
(1)找出数组中可能的三个数的组合
(2)计算它们的和与target的差值的绝对值
(3)排序,输出使得(2)中值最小时的和
class Solution:
def threeSumClosest(self, nums: List[int], target: int) -> int:
if len(nums) == 3:
return sum(nums)
result = {}
for i in range(len(nums)):
for j in range(i+1,len(nums)):
if j + 1 == len(nums):
break
for k in range(j+1,len(nums)):
add = nums[i] + nums[j] + nums[k]
result[add] = abs(add - target)
result = sorted(result.items(),key=lambda item:item[1])
print(result)
return result[0][0]
因为有三层循环,所以时间复杂度是O(n^3),提交之后超时。
解题思路二
参考2、3、4数之和的解法,上指针。
(1)固定一个元素,设置左指针和右指针
(2)当前三数和与目标最接近时,更新三数和为当前三数和
(3)当前三数和与目标不接近,如果比目标大,右指针左移;如果比目标小,左指针右移
(4) 当前三数和等于目标,输出目标/当前三数和
class Solution:
def threeSumClosest(self, nums: List[int], target: int) -> int:
if len(nums) == 3:
return sum(nums)
nums.sort()
mini = nums[0] + nums[1] + nums[2]
for i in range(len(nums)):
left = i + 1
right = len(nums) - 1
while left < right:
curr = nums[i] + nums[left] + nums[right]
if abs(curr-target) < abs(mini-target):
mini = curr
if curr > target:
right -= 1
elif curr < target:
left += 1
else:
return curr
return mini
如果你对之前的2、3、4数之和比较了解,那么这道题理论上是不会有太大的阻碍的。
后续可以考虑更优的解法,如何引入字典?
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/3sum-closest