77. 组合(Python)

给定两个整数 n 和 k，返回 1 … n 中所有可能的 k 个数的组合。

示例:

输入: n = 4, k = 2
输出:
[
[2,4],
[3,4],
[2,3],
[1,2],
[1,3],
[1,4],
]
题目就是要求我们从1到n找长度为k的组合数，所以回溯的终止条件就有了。

class Solution:
    def combine(self, n: int, k: int) -> list:
        ans = list()

        def backtrack(tmp: list, index: int) -> None:
            if len(tmp) == k:
                ans.append(tmp[:])
                return
            for i in range(index, n + 1):
                tmp.append(i)
                backtrack(tmp, i + 1)
                tmp.pop()
        backtrack([], 1)
        return ans