【回溯】LeetCode 77. Combinations

给定两个整数 n 和 k，返回 1 … n 中所有可能的 k 个数的组合。
示例: 输入: n = 4, k = 2 输出: [[2, 4], [3, 4], [2, 3], [1, 2], [1, 3], [1, 4]]

ans = []  # 用于存放总的返回结果
path = []  # 存放k个节点的组合


def backtracking(n, k, start_index):
    """
    :param start_index:存放每次回溯时要搜索的起始位置
    """
    # 终止条件
    if len(path) == k:
        ans.append(path[:])  # 收集结果
        return

    # 单层搜索
    for i in range(start_index, n+1):
        path.append(i)
        backtracking(n, k, i + 1)
        path.pop()


backtracking(4, 2, 1)
print(ans)