回溯理论基础
回溯法,一般可以解决如下几种问题:
- 组合问题:N个数里面按一定规则找出k个数的集合
- 切割问题:一个字符串按一定规则有几种切割方式
- 子集问题:一个N个数的集合里有多少符合条件的子集
- 排列问题:N个数按一定规则全排列,有几种排列方式
- 棋盘问题:N皇后,解数独等
回溯模板
void backtracking(参数) {
if (终止条件) {
存放结果;
return;
}
for (选择:本层集合中元素(树中节点孩子的数量就是集合的大小)) {
处理节点;
backtracking(路径,选择列表); // 递归
回溯,撤销处理结果
}
}
77. 组合
思路图
class Solution {
List<List<Integer>> res = new ArrayList<>();
LinkedList<Integer> path = new LinkedList<>();//用LinkedList,可以弹出最后一个元素
public List<List<Integer>> combine(int n, int k) {
backtrack(n,k,1);
return res;
}
public void backtrack(int n,int k,int startIndex){
if(path.size() == k){
res.add(new ArrayList<>(path));
return;
}
for(int i = startIndex;i <= n;i++){
path.add(i);
backtrack(n,k,i+1);
path.removeLast();
}
}
}