前言
我们社区陆续会将顾毅(Netflix 增长黑客,《iOS 面试之道》作者,ACE 职业健身教练。微博:@故胤道长[1])的 Swift 算法题题解整理为文字版以方便大家学习与阅读。
LeetCode 算法到目前我们已经更新了 15 期,我们会保持更新时间和进度(周一、周三、周五早上 9:00 发布),每期的内容不多,我们希望大家可以在上班路上阅读,长久积累会有很大提升。
不积跬步,无以至千里;不积小流,无以成江海,Swift社区 伴你前行。如果大家有建议和意见欢迎在文末留言,我们会尽力满足大家的需求。
难度水平:中等
1. 描述
给你一个长度为 n 的整数数组 nums 和 一个目标值 target。请你从 nums 中选出三个整数,使它们的和与 target 最接近。
返回这三个数的和。
假定每组输入只存在恰好一个解。
2. 示例
示例 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
3. 答案
- class ThreeSum {
- func threeSum(_ nums: [Int]) -> [[Int]] {
- var res = [[Int]]()
- guard nums.count >= 3 else {
- return res
- }
- let nums = nums.sorted()
- for i in 0..<nums.count - 2 {
- if i > 0 && nums[i] == nums[i - 1] {
- continue
- }
- let firstNum = nums[i], remainingSum = -firstNum
- var m = i + 1, n = nums.count - 1
- while m < n {
- if nums[m] + nums[n] == remainingSum {
- res.append([firstNum, nums[m], nums[n]])
- repeat {
- m += 1
- } while nums[m] == nums[m - 1] && m < n
- repeat {
- n -= 1
- } while nums[n] == nums[n + 1] && m < n
- } else if nums[m] + nums[n] < remainingSum {
- m += 1
- } else {
- n -= 1
- }
- }
- }
- return res
- }
- }
- 主要思想:对数组进行排序并遍历,根据它们的和大于或不大于目标,向左递增或向右递减
- 时间复杂度: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