赞
踩
Day 24
回溯 : 基础知识、 组合
回溯法→回溯搜索法
for循环:横向遍历
递归:纵向遍历
回溯法,一般可以解决如下几种问题:
void backtracking(参数) {
if (终止条件) {
存放结果;
return;
}
for (选择:本层集合中元素(树中节点孩子的数量就是集合的大小)) {
处理节点;
backtracking(路径,选择列表); // 递归
回溯,撤销处理结果
}
}
给定两个整数 n 和 k,返回范围 [1, n] 中所有可能的 k 个数的组合。
你可以按 任何顺序 返回答案。
示例 :
输入:n = 4, k = 2
输出:
[
[2,4],
[3,4],
[2,3],
[1,2],
[1,3],
[1,4],
]
class Solution { public: // 存放符合条件结果的集合 vector<vector<int>> result; // 用来存放符合条件结果 vector<int> path; void backtracking(int n, int k, int startIndex) { // 终止条件 if (path.size() == k) { result.push_back(path); return ; } // 循环(横向处理) for (int i = startIndex; i <= n; i++) { // 处理节点 path.push_back(i); // 递归(纵向处理) backtracking(n, k, i+1); // 回溯,撤销处理的节点 path.pop_back(); } return ; } vector<vector<int>> combine(int n, int k) { backtracking(n, k, 1); return result; } };
剪枝优化
for (int i = startIndex; i <= n - (k - path.size()) + 1; i++) // i为本次搜索的起始位置
补充解释:
path.size() : 已经找的个数
k-path.size() :还需找的个数
【x, n】的数组长度起码应该是k-path.size()才有继续搜索的可能, 那么就有 n-x+1 = k-path.size() , 解方程得 x = n+1 - (k-path.size())
而且这个x是可以作为起点往下搜的 也就是for(i = startIndex; i<=x; i++) 这里的x是可以取到的
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。