77. 组合

题目-中等难度

给定两个整数 n 和 k，返回范围 [1, n] 中所有可能的 k 个数的组合。

你可以按 任何顺序 返回答案。

示例

示例 1：

输入：n = 4, k = 2
输出：
[
[2,4],
[3,4],
[2,3],
[1,2],
[1,3],
[1,4],
]

示例 2：

输入：n = 1, k = 1
输出：[[1]]

提示：

• 1 <= n <= 20
• 1 <= k <= n

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/combinations
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

回溯

class Solution:
    def combine(self, n: int, k: int) -> List[List[int]]:
        li = []  # 用于存储所有组合的列表。
        lli = [] # 临时列表，用于存储当前组合。

        # 回溯函数。
        def backtrack(index):
            # 如果当前组合的长度等于k，则找到一个有效组合。
            if len(lli) == k:
                li.append(lli.copy()) # 将当前组合复制并添加到最终列表。
                return

            # 遍历从index到n的所有数字。
            for i in range(index, n):
                lli.append(i + 1) # 将数字添加到当前组合。
                backtrack(i + 1)  # 递归调用，移动到下一个数字。
                lli.pop()         # 删除最后一个数字以进行回溯。

        backtrack(0) # 从数字1开始回溯。
        return li