1. 程式人生 > >最接近的三數之和 leetcode

最接近的三數之和 leetcode

給定一個包括 n 個整數的陣列 nums和 一個目標值 target。找出 nums中的三個整數,使得它們的和與 target 最接近。返回這三個數的和。假定每組輸入只存在唯一答案。

例如,給定陣列 nums = [-1,2,1,-4], 和 target = 1.

與 target 最接近的三個數的和為 2. (-1 + 2 + 1 = 2).
class Solution:
    def threeSumClosest(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: int
        """
        result=nums[0]+nums[1]+nums[2]
        nums.sort()
        for i in range(len(nums)-2):
            if i>0 and nums[i]==nums[i-1]:
                continue
            l,r=i+1,len(nums)-1
            while l<r:
                s=nums[i]+nums[l]+nums[r]
                if s==target:
                    return s
                if abs(s-target)<abs(result-target):
                    result=s
                elif s<target:
                    l += 1
                    while l < r and nums[l] == nums[l-1]:
                        l+=1
                else:
                    r -= 1
                    while l < r and nums[r]==nums[r+1]:
                        r-=1
        return result