回溯算法是一种非常实用的算法解题思路,LeetCode中题库中有106道回溯的试题,而且难度都在中等以上,掌握回溯算法对高效快速解决很多问题都很有帮助。今天我们主要看看回溯算法的解题套路和实用回溯解决排列组合问题的题解。
1.回溯算法特点
回溯算法过程:在搜索中寻找问题的解,发现不满足条件时,就回溯返回上一步,尝试别的路径。
从上面的定义可以看出,回溯算法依赖于递归实现,与递归算法的差异就在于一个“回”字,即修改了状态并做了尝试之后,需要回过来把状态改回去再做其他尝试,通过多轮回溯逐步求出所有的解。
回溯算法的应用:回溯算法特别使用用于有排列组合或者路径探索的场景。
2.使用回溯解决组合问题
2.1 n个数中的k个组合
77. 组合
剑指 Offer II 080. 含有 k 个元素的组合
题目描述:给定两个整数 n 和 k,返回范围 [1, n] 中所有可能的 k 个数的组合。
先上源码如下:
class Solution {
public:
vector<vector<int>> combine(int n, int k) {
backtrack(n, k, 1);
return res;
}
private:
vector<vector<int>> res;
vector<int> temp;
void backtrack(int n, int k, int index){
if(temp.size() == k){ //递归结束条件
res.push_back(temp);
return;
}
for(int i = index; i <= n && temp.size() < k; i++){
temp.push_back(i); //修改状态
backtrack(n, k, i+1); //递归回溯
temp.pop_back(); //改回状态
}
}
};
解题分析:
- 定义变量:排列组合问题一般会返回一个二维数组的结果,这里定义了成员变量vector<vector> res,还会需要一个探索过程中的一维数组临时变量vector temp。
- 定义回溯函数:回溯函数依赖递归实现,入口函数不方便递归,一般都写一个递归子函数backtrack来完成。
- 定义回溯函数参数列表:一般是入口函数参数列表+ 递归索引位置(组合问题用来取不