日新刷题 - 16. 最接近的三数之和

108 阅读1分钟

持续创作,加速成长!这是我参与「掘金日新计划 · 6 月更文挑战」的第9天,点击查看活动详情

一 描述

16. 最接近的三数之和 - 力扣(LeetCode)

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

返回这三个数的和。

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

 

示例 1:

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

示例 2:

输入:nums = [0,0,0], target = 1
输出:0

 

提示:

  • 3 <= nums.length <= 1000
  • -1000 <= nums[i] <= 1000
  • -10^4 <= target <= 10^4

二 分析

  • 对数组进行排序
  • 初始化三元素之和 total = nums[0]+nums[1]+nums[2],由于target不确定,total不能设定为0或-1
  • 双指针流程:
    • 若nums[i] > target,退出循环:排序之后,后面的元素(nums[i+1]和nums[i+2])都大于等于nums[i],因此结果不可能等于target
    • 令左指针left = i+1,右指针right = length-1,执行循环,判断选取的三元素之和total与target之间的关系,移动左右指针,以寻找最优结果。
    • cha用来存储total - target的最小绝对值(当前状态下),完成所有遍历后,最小绝对值所对应的元素和为最终结果
      • 若 abs(total-target) < cha,更新cha和res
      • 当total - target > 0时,right = right - 1
      • 当total - target < 0时,left = left + 1
      • 当total - target = 0时,所取三元素之和等于target,可以直接返回结果target
    • 遍历所有完成,返回res

三 答案

class Solution:
    def threeSumClosest(self, nums: List[int], target: int) -> int:
        nums.sort()
        length = len(nums)
        total = nums[0]+nums[1]+nums[2]
        cha = abs(total - target)
        res = total

        for i in range(length-2):#第一层循环,固定其中一个元素
            if nums[i] > target:
                break
            left = i+1
            right = length-1 
            while(left < right):#使用双指针
                total = nums[i]+nums[left]+nums[right]
                if abs(total-target) < cha:
                    cha = abs(total - target)
                    res = total
                if total - target > 0:
                    right = right - 1
                elif total - target < 0:
                    left = left + 1 
                else:
                    return target
        return res

四 参考

双指针优化到极致 - 最接近的三数之和 - 力扣(LeetCode)

双指针java - 最接近的三数之和 - 力扣(LeetCode)