1. 程式人生 > 其它 >16. 最接近的三數之和 (leetcode)

16. 最接近的三數之和 (leetcode)

技術標籤:leetcodeleetcodepython演算法

題目:

給定一個包括 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

來源:https://leetcode-cn.com/problems/3sum-closest

打卡:(方法還是排序+三索引)

class Solution:
    def threeSumClosest(self, nums: List[int], target: int) -> int:
        if len(nums)<3:
            return []
        nums_sorted=sorted(nums)
        result_closet=sum(nums[0:3])
        gap = float("inf"
) for i in range(len(nums_sorted)-2): l=i+1 r=len(nums)-1 while l<r: s=nums_sorted[i]+nums_sorted[l]+nums_sorted[r] if abs(s-target) < gap : result_closet = s gap = abs(s-target)
if s-target < 0: l+=1 elif s-target > 0: r-=1 else: return result_closet return result_closet