每次从集合中选取元素,可选择的范围随着选择的进行而收缩,调整可选择的范围。
图中可以发现n相当于树的宽度,k相当于树的深度。
那么如何在这个树上遍历,然后收集到我们要的结果集呢?
图中每次搜索到了叶子节点,我们就找到了一个结果。
相当于只需要把达到叶子节点的结果收集起来,就可以求得 n个数中k个数的组合集合。
77题 组合
class Solution(object): def combine(self, n, k): a = self.backtracking(n, k, 1, [], []) return a def backtracking(self, n, k, start_index, path, result): if len(path) == k: result.append(path[:]) return for i in range(start_index, n-(k-len(path))+2): path.append(i) self.backtracking(n, k, i + 1, path, result) path.pop() return result
【思考】此题相当于一个回溯算法的模板,剪枝为难点,其实可理解为每次循环的有边界。