美文网首页
[LeetCode]16、最接近的三个数之和

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

作者: 河海中最菜 | 来源:发表于2019-07-27 14:57 被阅读0次

题目描述

给定一个包括 n 个整数的数组 nums 和 一个目标值 target。找出 nums 中的三个整数,使得它们的和与 target 最接近。返回这三个数的和。假定每组输入只存在唯一答案。

例如,给定数组 nums = [-1,2,1,-4], 和 target = 1.

与 target 最接近的三个数的和为 2. (-1 + 2 + 1 = 2).

思路

设置一个返回结果,用于更新最接近的目标的三个数之和
同样的操作,固定一个数,剩下来的二分查找

  • 首先进行数组排序,时间复杂度 O(nlogn)
  • 在数组 nums 中,进行遍历,每遍历一个值利用其下标i,形成一个固定值 nums[i]
  • 再使用前指针指向 start = i + 1 处,后指针指向 end = nums.length - 1 处,也就是结尾处
  • 根据 sum = nums[i] + nums[start] + nums[end] 的结果,判断 sum 与目标 target 的距离,如果更近则更新结果 ans,同时判断 sum 与 target 的大小关系,因为数组有序,如果 sum > target 则 end--,如果 sum < target 则 start++,如果 sum == target 则说明距离为 0 直接返回结果
    整个遍历过程,固定值为 n 次,双指针为 n 次,时间复杂度为 O(n^2)
    总时间复杂度:O(nlogn) + O(n^2) = O(n 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

相关文章

网友评论

      本文标题:[LeetCode]16、最接近的三个数之和

      本文链接:https://www.haomeiwen.com/subject/xmkerctx.html