LeetCode - #216 组合总和 III

Posted 2022-10-24

前言

我们社区陆续会将顾毅（Netflix 增长，《iOS 面试之道》作者，ACE 职业健身教练。）的 Swift 算法题题解整理为文字版以方便大家学习与阅读。

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

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

难度水平：中等

1. 描述

找出所有相加之和为 ​​n​​​ 的 ​​k​​ 个数的组合，且满足下列条件：

• 只使用数字 1 到 9
• 每个数字 最多使用一次

返回 所有可能的有效组合的列表 。该列表不能包含相同的组合两次，组合可以以任何顺序返回。

2. 示例

示例 1

输入: k = 3, n = 7
输出: [[1,2,4]]
解释:
1 + 2 + 4 = 7

示例 2

输入: k = 3, n = 9
输出: [[1,2,6], [1,3,5], [2,3,4]]
解释:
1 + 2 + 6 = 9
1 + 3 + 5 = 9
2 + 3 + 4 = 9

示例 3

输入: k = 4, n = 1
输出: []
解释: 不存在有效的组合。
在[1,9]范围内使用4个不同的数字，我们可以得到的最小和是1+2+3+4 = 10，因为10 > 1，没有有效的组合。

提示：

• ​​2 <= k <= 9​​
• ​​1 <= n <= 60​​

3. 答案

class CombinationSumIII 
    func combinationSum3(k: Int, _ n: Int) -> [[Int]] 
        let candidates = [Int](1...9)
        var res = [[Int]](), path = [Int]()

        dfs(&res, &path, candidates, n, 0, k)

        return res
    

    fileprivate func dfs(_ res: inout [[Int]], _ path: inout [Int], _ candidates: [Int], _ target: Int, _ index: Int, _ size: Int) 
        if target == 0 && path.count == size 
            res.append(Array(path))
            return
        

        guard path.count < size else 
            return
        

        for i in index..<candidates.count 
            let candidate = candidates[i]

            guard candidate <= target else 
                break
            

            path.append(candidate)
            _dfs(&res, &path, candidates, target - candidate, i + 1, size)
            path.removeLast()
        
    

• 主要思想：经典的深度优先搜索。
• 时间复杂度：O(2^n)
• 空间复杂度：O(n)

该算法题解的仓库：LeetCode-Swift

点击前往 LeetCode 练习

关于我们

我们是由 Swift 爱好者共同维护，我们会分享以 Swift 实战、SwiftUI、Swift 基础为核心的技术内容，也整理收集优秀的学习资料。