C++之回溯算法解决子集、全排列和组合问题

这篇文章回溯算法团灭子集、排列、组合问题写得很好,在此记录一下自己的理解以及使用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
  • 5
    点赞
  • 19
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值