这篇文章回溯算法团灭子集、排列、组合问题写得很好,在此记录一下自己的理解以及使用LeetCode真题来帮助更好地运用。
先说一句话:回溯算法就类似于多叉树的遍历问题!!!!
作者说回溯算法就是多叉树的遍历问题,我认为其中还有一个区别,那就是遍历问题每个元素只会使用一次,而回溯算法中,每个元素使用多次,这也是为何算法中会有撤销选择这一步了。。
上面这句话猛一看很难理解,接下来咱们慢慢揭开它的神秘面纱!!
一、从二叉树的遍历开始讲起
我之前写过c++之二叉树的递归实现与广义表实现,其中关于二叉树的递归遍历解法,我们可以用以下伪代码进行概括:
void traverse(TreeNode root) {
// 在此写前序遍历相关操作
traverse(root.left)
// 在此写中序遍历相关操作
traverse(root.right)
// 在此写后序遍历相关操作
}
实际过程如下:
前序遍历的代码在进入某一个节点之前的那个时间点执行,后序遍历代码在离开某个节点之后的那个时间点执行。
二、引入回溯算法
先来看一个简单的全排列问题,来了解回溯算法
比方说给三个数 [1,2,3],找出它的所有全排列组合。。
我们的解法:先固定第一位为 1,然后第二位可以是 2,那么第三位只能是 3;然后可以把第二位变成 3,第三位就只能是 2 了;然后就只能变化第一位,变成 2,然后再穷举后两位……
其实这就是回溯算法,我们高中无师自通就会用,或者有的同学直接画出如下这棵回溯树:
我们不妨把这棵树称为回溯算法的「决策树」。
为啥说这是决策树呢,因为你在每个节点上其实都在做决策。比如说你站在上图的红色节点上:
你现在就在做决策,可以选择 1 那条树枝,也可以选择 3 那条树枝。为啥只能在 1 和 3 之中选择呢?因为 2 这个树枝在你身后,这个选择你之前做过了,而全排列是不允许重复使用数字的。
在此,我们可以总结一下回溯算法的基本思想:
result = []
void backtrack(路径, 选择列表):
if 满足结束条件:
result.add(路径)
return
for 选择 in 选择列表:
做选择
backtrack(路径, 选择列表)
撤销选择
1、路径:也就是已经做出的选择。
2、选择列表:也就是你当前可以做的选择。
3、结束条件:也就是到达决策树底层,无法再做选择的条件。
其核心就是 for 循环里面的递归,在递归调用之前「做选择」,在递归调用之后「撤销选择」,特别简单
这也就是多叉树(决策树的遍历),因为每个元素要使用多次,所以我们要有撤销选择这一步。这也是与遍历的区别。。
三、解决子集问题
Leetcode78. 子集
给定一组不含重复元素的整数数组 nums,返回该数组所有可能的子集(幂集)。
输入: nums = [1,2,3]
输出: [ [3], [1], [2], [1,2,3], [1,3], [2,3], [1,2], [] ]
class Solution {
public:
vector<vector<int>> res;
void dfs(vector<int>& nums,int start,vector<int>& track)
{
//满足结束条件
res.push_back(track);
for(int i=start;i<nums.size();i++)
{
//做出选择nums[i],将其放入路径中
track.push_back(nums[i]);
//递归,填下一个数
dfs(nums,i+1,track);
//撤销选择
track.pop_back();
}
return ;
}
vector<vector<int>> subsets(vector<int>& nums) {
vector<int> track;
dfs(nums,0,track);
return res;
}
};
Leetcode90. 子集 II
给定一个可能包含重复元素的整数数组 nums,返回该数组所有可能的子集(幂集)。
输入: [1,2,2]
输出: [ [2], [1], [1,2,2], [2,2], [1,2], [] ]
class Solution {
public:
vector<vector<int>> res;
void dfs(vector<int>& nums,int start,vector<int>& track)
{
res.push_back(track);
for(int i=start;i<nums.size();i++)
{
track.push_back(nums[i]);
dfs(nums,i+1,track);
track.pop_back();
}
return ;
}
vector<vector<int>> subsetsWithDup(vector<int>& nums) {
vector<int> track;
//注意这一步,不然有可能得出混乱的结果
sort(nums.begin(),nums.end());
//先解设元素都不相同
dfs(nums,0,track);
//删除重复的结果
sort(res.begin(),res.end());
res.erase