LeetCode回溯模板，77组合

回溯模板

void backtracking(参数) {
    if (终止条件) {
        存放结果;
        return;
    }

    for (选择：本层集合中元素（树中节点孩子的数量就是集合的大小）) {
        处理节点;
        backtracking(路径，选择列表); // 递归
        回溯，撤销处理结果
    }
}

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

c++代码实现

class Solution {
public:
    vector<vector<int>> results;
    vector<int> path;

    void backtracking(int n, int k, int startIndex) {
        // 满足k条件
        if (path.size() == k) {
            results.push_back(path);
            return;
        }

        for (int i = startIndex; i <= n - (k-path.size()) + 1; i++) {
            path.push_back(i);          // 处理节点
            backtracking(n, k, i+1);    // 递归
            path.pop_back();            // 回溯
        }

    }

    vector<vector<int>> combine(int n, int k) {
        // [1, n]
        backtracking(n, k, 1);
        return results;
    }
};

python 代码实现

class Solution:
    def combine(self, n: int, k: int) -> List[List[int]]:
        results = []
        path = []

        def backtracking(n, k, startIndex):
            if len(path) == k:
                results.append(path[:])
                return
            
            # 剪枝， 最后k - len(path)个节点直接构造结果，无需递归
            last_startidx = n - (k - len(path)) + 1
            results.append(path + [idx for idx in range(last_startidx, n + 1)])


            for i in range(startIndex, last_startidx):
                path.append(i)
                backtracking(n, k, i+1)
                path.pop()

        backtracking(n, k, 1)
        return results