LeetCode 之最接近的三数之和

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

[[441825]]

前言

我们社区陆续会将顾毅(Netflix 增长黑客,《iOS 面试之道》作者,ACE 职业健身教练。微博:@故胤道长[1])的 Swift 算法题题解整理为文字版以方便大家学习与阅读。

LeetCode 算法到目前我们已经更新了 15 期,我们会保持更新时间和进度(周一、周三、周五早上 9:00 发布),每期的内容不多,我们希望大家可以在上班路上阅读,长久积累会有很大提升。

不积跬步,无以至千里;不积小流,无以成江海,Swift社区 伴你前行。如果大家有建议和意见欢迎在文末留言,我们会尽力满足大家的需求。

难度水平:中等

1. 描述

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

返回这三个数的和。

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

2. 示例

示例 1

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

示例 2

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

约束条件:

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

3. 答案

  1. class ThreeSum { 
  2.     func threeSum(_ nums: [Int]) -> [[Int]] { 
  3.         var res = [[Int]]() 
  4.          
  5.         guard nums.count >= 3 else { 
  6.             return res 
  7.         } 
  8.          
  9.         let nums = nums.sorted() 
  10.          
  11.         for i in 0..<nums.count - 2 { 
  12.             if i > 0 && nums[i] == nums[i - 1] { 
  13.                 continue 
  14.             } 
  15.              
  16.             let firstNum = nums[i], remainingSum = -firstNum 
  17.             var m = i + 1, n = nums.count - 1 
  18.              
  19.             while m < n { 
  20.                 if nums[m] + nums[n] == remainingSum { 
  21.                     res.append([firstNum, nums[m], nums[n]]) 
  22.                      
  23.                     repeat { 
  24.                         m += 1 
  25.                     } while nums[m] == nums[m - 1] && m < n 
  26.                      
  27.                     repeat { 
  28.                         n -= 1 
  29.                     } while nums[n] == nums[n + 1] && m < n 
  30.                 } else if nums[m] + nums[n] < remainingSum { 
  31.                     m += 1 
  32.                 } else { 
  33.                     n -= 1 
  34.                 } 
  35.             } 
  36.         } 
  37.          
  38.         return res 
  39.     } 
  • 主要思想:对数组进行排序并遍历,根据它们的和大于或不大于目标,向左递增或向右递减
  • 时间复杂度:O(n^2)
  • 空间复杂度:O(nC3)

该算法题解的仓库:LeetCode-Swift[2]

点击前往 LeetCode[3] 练习

参考资料

[1]@故胤道长:

https://m.weibo.cn/u/1827884772

[2]LeetCode-Swift:

https://github.com/soapyigu/LeetCode-Swift

[3]LeetCode:

https://leetcode.com/problems/3sum-closest

 

责任编辑:姜华 来源: Swift社区
相关推荐

2021-12-31 09:01:44

LeetCode 罗马数字四数之和

2021-04-30 16:04:03

PythonNumPy元组

2021-12-17 08:51:41

LeetCode最长公共前缀算法

2013-08-20 13:34:02

创业极客

2021-09-11 16:42:26

AndroidAndroid 12

2009-02-05 08:23:15

裁员计算机业

2023-08-31 21:44:07

2021-03-01 14:47:00

Linux LiteWindows发行版

2020-10-12 11:16:32

数组特定值元素

2024-03-22 13:05:23

数据训练

2020-07-01 14:07:04

阿里区块链支付宝

2022-02-15 08:25:22

hash表快排二分查找

2012-05-15 16:00:35

小米

2023-09-19 09:22:54

数据训练

2021-06-30 14:38:20

5G 5G网络GSMA

2009-04-10 09:04:41

Windows 7微软操作系统

2016-04-11 18:14:07

2021-03-25 16:50:30

智能医院物联网人工智能

2011-11-21 09:55:32

NFC

2018-03-12 09:05:48

高并发微服务架构
点赞
收藏

51CTO技术栈公众号