力扣216题详解：组合总和 III 的多种解法与模拟面试

最新推荐文章于 2024-11-14 22:14:16 发布

数据分析螺丝钉

最新推荐文章于 2024-11-14 22:14:16 发布

阅读量238

点赞数 1

分类专栏： LeetCode刷题与模拟面试文章标签：面试算法 leetcode 经验分享 python

本文链接：https://blog.csdn.net/CCIEHL/article/details/140893560

版权

LeetCode刷题与模拟面试专栏收录该内容

224 篇文章 74 订阅

订阅专栏

在本篇文章中，我们将详细解读力扣第216题“组合总和 III”。通过学习本篇文章，读者将掌握如何使用多种方法来解决这一问题，并了解相关的复杂度分析和模拟面试问答。每种方法都将配以详细的解释，以便于理解。

问题描述

力扣第216题“组合总和 III”描述如下：

找出所有相加之和为 n 的 k 个数的组合。组合中只允许含有 1 - 9 的正整数，并且每种组合中不存在重复的数字。

说明：

所有数字都是正整数。
解集不能包含重复的组合。

示例:
输入: k = 3, n = 7
输出: [[1,2,4]]
示例:
输入: k = 3, n = 9
输出: [[1,2,6], [1,3,5], [2,3,4]]

解题思路

方法一：回溯法

初步分析：
- 使用回溯法来遍历所有可能的组合，找到满足条件的组合。
步骤：
- 定义一个辅助函数 backtrack，接受当前组合、起始数字、剩余数字个数和剩余总和作为参数。
- 在辅助函数中，判断当前组合是否满足条件，如果满足则加入结果集。
- 递归调用辅助函数，尝试加入下一个数字，直到满足条件或无法继续。

代码实现

def combinationSum3(k, n):
    def backtrack(start, comb, k, n):
        if k == 0 and n == 0:
            result.append(list(comb))
            return
        if k == 0 or n < 0:
            return
        for i in range(start, 10):
            comb.append(i)
            backtrack(i + 1, comb, k - 1, n - i)
            comb.pop()
    
    result = []
    backtrack(1, [], k, n)
    return result

# 测试案例
print(combinationSum3(3, 7))  # 输出: [[1,2,4]]
print(combinationSum3(3, 9))  # 输出: [[1,2,6], [1,3,5], [2,3,4]]

方法二：迭代法（仅作为补充）

初步分析：
- 使用迭代法生成所有可能的组合，检查每个组合是否满足条件。
步骤：
- 生成所有可能的组合。
- 检查每个组合是否满足条件，满足条件则加入结果集。

代码实现

from itertools import combinations

def combinationSum3(k, n):
    result = []
    for comb in combinations(range(1, 10), k):
        if sum(comb) == n:
            result.append(list(comb))
    return result

# 测试案例
print(combinationSum3(3, 7))  # 输出: [[1,2,4]]
print(combinationSum3(3, 9))  # 输出: [[1,2,6], [1,3,5], [2,3,4]]

复杂度分析

时间复杂度：
- 回溯法：O(C(9, k))，其中 C(9, k) 是从 9 个数中选出 k 个数的组合数。
- 迭代法：O(C(9, k))，需要生成所有可能的组合并检查每个组合。
空间复杂度：
- 回溯法：O(k)，用于递归调用栈和当前组合。
- 迭代法：O(C(9, k))，用于存储所有可能的组合。