深圳幻海软件技术有限公司 欢迎您!

LeetCode 之最接近的三数之和

2023-02-28

前言我们社区陆续会将顾毅(Netflix增长黑客,《iOS面试之道》作者,ACE职业健身教练。微博:@故胤道长[1])的Swift算法题题解整理为文字版以方便大家学习与阅读。LeetCode算法到目前我们已经更新了15期,我们会保持更新时间和进度(周一、周三、周五早上9:00发布),每期的内容不多,

前言

我们社区陆续会将顾毅(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) 。 
  • 1.
  • 2.
  • 3.

示例 2

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

约束条件:

  • 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 
    } 

  • 1.
  • 2.
  • 3.
  • 4.
  • 5.
  • 6.
  • 7.
  • 8.
  • 9.
  • 10.
  • 11.
  • 12.
  • 13.
  • 14.
  • 15.
  • 16.
  • 17.
  • 18.
  • 19.
  • 20.
  • 21.
  • 22.
  • 23.
  • 24.
  • 25.
  • 26.
  • 27.
  • 28.
  • 29.
  • 30.
  • 31.
  • 32.
  • 33.
  • 34.
  • 35.
  • 36.
  • 37.
  • 38.
  • 39.
  • 40.
  • 主要思想:对数组进行排序并遍历,根据它们的和大于或不大于目标,向左递增或向右递减
  • 时间复杂度: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