0
点赞
收藏
分享

微信扫一扫

最接近的三数之和Python解法

西街小学的王 2022-02-07 阅读 79

给你一个长度为 n 的整数数组 nums 和 一个目标值 target。请你从 nums 中选出三个整数,使它们的和与 target 最接近。

返回这三个数的和。

假定每组输入只存在恰好一个解。

例:

输入:nums = [-1,2,1,-4], target = 1
输出:2
解释:与 target 最接近的和是 2 (-1 + 2 + 1 = 2) 。

解析:双指针,数组从小到大排序,左指针从最左开始,又指针从右开始,然后便利中间的值进行相加,当三数之和小于target时,中间的值右移一个值,当三数之和大于target时,右指针左移一个值,且左指针需要小于右指针。

class Solution(object):
    def threeSumClosest(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: int
        """
        nums.sort()
        n = len(nums)
        nearest = 10 ** 7

        for i in range(n - 2):
            lo, hi = i + 1, n - 1
            # 双指针
            while lo < hi:
                total = nums[i] + nums[lo] + nums[hi]
                if total == target:
                    return total
                elif total < target:
                    lo += 1
                else:
                    hi -= 1  # 右指针左移
                # 取绝对值更小的那个,也更接近
                nearest = min(nearest, total, key=lambda x: abs(x - target))
        
        return nearest
举报

相关推荐

0 条评论