[LeetCode]16、最接近的三个数之和

2019-07-27  本文已影响0人  河海中最菜

题目描述

给定一个包括 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):
        if not nums or len(nums) < 3:
            return None
        nums.sort()
        ans = nums[0] + nums[1] + nums[2]
        for i in range(len(nums) - 2):
            left, right = i + 1, len(nums) - 1
            while left < right:
                sums = nums[i] + nums[left] + nums[right]
                if abs(target - ans) > abs(target - sums):
                    ans = sums
                if sums > target:
                    right -= 1
                elif sums < target:
                    left += 1
                else:
                    return ans
        return ans

AC16
上一篇下一篇

猜你喜欢

热点阅读