什么是回溯算法
回溯算法一般跟递归一起出现,是用来解决排列组合,分割字符串等问题的一种算法。实际上是一种暴力算法。
例 : leetcode上的77题
给定两个整数 n
和 k
,返回范围 [1, n]
中所有可能的 k
个数的组合。
你可以按 任何顺序 返回答案。
这里我们假设n=4,k=2。 那么答案就是(1,2),(1,3), (1,4),(2,3),(2,4),
(3,4)这6种情况
按常规思路我们就直接来两个for循环解决了,但是如果k是10,100或者更大呢,那样的话再用for来解决就不太现实了,这就需要用到回溯了。
这种问题可以用树来理解,首先我们选取一个根节点1,然后不断往下继续搜索,当搜索的数量达到2也就是题中的k时,我们便可以停下来把此时得到的一组数据存起来,然后返回到上面继续搜索。其中回溯的作用就是得到我们要的数据后把最后得到的那个节点上的数挪出去。比如我们得到(1,2)了,然后回溯把2挪出去,1继续和3匹配得到(1,3)再把3挪出去,再得到(1,4)....
下面为回溯算法的基本模板:
首要要有个终止条件
如上面的例题,当我们收集的数据个数达到k时就停止
if( temp.size() == k)
{
res.push_back(temp);
return;
}
之后就是一层for循环不断的遍历节点
for i index i<=n i++
temp.push_back(i); // 把i存入数组
backtracking(i+1,..); // 找下一个节点
temp.pop_back(); //回溯
一般的回溯算法的模板就是这样,下面为整体函数代码
class Solution {
public:
vector<vector<int>> res;
vector<int> r;
void backracking(int startindex,int k,int n)
{
if(r.size() == k)
{
res.push_back(r);
return;
}
for(int i = startindex; i<=n; i++)
{
r.push_back(i);
backracking(i+1,k,n);
r.pop_back();
}
}
vector<vector<int>> combine(int n, int k) {
backracking(1,k,n);
return res;
}
};
我描述的也不是很清楚,还是得自己多在纸上写写画画,多理解理解,不懂就在纸上照着代码写一遍,把数据代入,也不算太难理解。