码迷,mamicode.com
首页 > 其他好文 > 详细

leetcode 16:最接近的三数之和

时间:2019-08-10 11:27:42      阅读:62      评论:0      收藏:0      [点我收藏+]

标签:ret   返回   排序   ble   leetcode   包括   输入   span   数组   

Problem:

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

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

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

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/3sum-closest

Answer:

1、暴力解法:时间复杂度O(n^3),超出时间限制。

 

2、O(n^2)的解法:

先将数组排序,对于每一个i从0到n - 1,令left = i + 1,right = size - 1,为了寻找更接近target的三数之和,

若三数之和大于target则让right - 1,若三数之和等于target则返回target,若三数之和小于target则让left + 1。

 

设排序后的数组arr为a, b, c, d, e, f。

令arr[i] = a, arr[left] = b, arr[right] = f。

若a + b + f < target,则令left + 1,即arr[left] = c,

b不再有与小于f的数(如e)组合的可能,但由于e < f,a + b + e < a + b + f,离target更远,因而不必考虑。

 

Code:

 1 class Solution:
 2     def threeSumClosest(self, nums: List[int], target: int) -> int:
 3         nums_len = len(nums)
 4         nums.sort()
 5         min_dist = float(inf)
 6         for i in range(nums_len):
 7             left = i + 1
 8             right = nums_len - 1
 9             while left < right:
10                 sum_ = nums[i] + nums[left] + nums[right]
11                 dist = abs(target - sum_)
12                 if dist < min_dist:
13                     min_dist = dist
14                     ans = sum_
15                 if sum_ == target:
16                     return target
17                 elif sum_ > target:
18                     right -= 1
19                 else:
20                     left += 1
21         return ans

 

leetcode 16:最接近的三数之和

标签:ret   返回   排序   ble   leetcode   包括   输入   span   数组   

原文地址:https://www.cnblogs.com/lxc1910/p/11330787.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!