[[441825]]

前言

我們社區(qū)陸續(xù)會(huì)將顧毅(Netflix 增長(zhǎng)黑客，《iOS 面試之道》作者，ACE 職業(yè)健身教練。微博:@故胤道長(zhǎng)[1])的 Swift 算法題題解整理為文字版以方便大家學(xué)習(xí)與閱讀。

LeetCode 算法到目前我們已經(jīng)更新了 15 期，我們會(huì)保持更新時(shí)間和進(jìn)度(周一、周三、周五早上 9:00 發(fā)布)，每期的內(nèi)容不多，我們希望大家可以在上班路上閱讀，長(zhǎng)久積累會(huì)有很大提升。

不積跬步，無(wú)以至千里;不積小流，無(wú)以成江海，Swift社區(qū) 伴你前行。如果大家有建議和意見(jiàn)歡迎在文末留言，我們會(huì)盡力滿足大家的需求。

難度水平：中等

1. 描述

給你一個(gè)長(zhǎng)度為 n 的整數(shù)數(shù)組 nums 和 一個(gè)目標(biāo)值 target。請(qǐng)你從 nums 中選出三個(gè)整數(shù)，使它們的和與 target 最接近。

返回這三個(gè)數(shù)的和。

假定每組輸入只存在恰好一個(gè)解。

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

• 主要思想：對(duì)數(shù)組進(jìn)行排序并遍歷，根據(jù)它們的和大于或不大于目標(biāo)，向左遞增或向右遞減
• 時(shí)間復(fù)雜度：O(n^2)
• 空間復(fù)雜度：O(nC3)

該算法題解的倉(cāng)庫(kù)：LeetCode-Swift[2]

點(diǎn)擊前往 LeetCode[3] 練習(xí)

參考資料

[1]@故胤道長(zhǎng):

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

[2]LeetCode-Swift:

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

[3]LeetCode:

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