基于剪枝的回溯法

Posted CUHK.CS我高攀不起

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了基于剪枝的回溯法相关的知识,希望对你有一定的参考价值。

主要内容

好久没上Leetcode,发现首页居然有一道中等题"全排列II"错了好多次一直没对,发现是带剪枝的回溯类型,做完了就稍微记录一下。

今天主要通过两个例题,一是回顾回溯法,二就是学会回溯的同时剪枝。

简单的全排列

力扣中的Q46,原题如下:

题目的温馨提示也很体贴地告诉了你数组中的各元素不相等,所以轮不到剪枝的技巧派上用场。

直接贴出代码:

class Solution 
public:
    vector<vector<int>> ans;
    vector<int> temp;
    void backtrack(vector<int>& nums, vector<bool>& traverse)
        if(temp.size() == nums.size())
            ans.push_back(temp);
            return;
        
        for(int i = 0 ; i < nums.size() ; i ++)
            if(!traverse[i])
                temp.push_back(nums[i]);
                traverse[i] =  true;
                backtrack(nums, traverse);
                traverse[i] = false;
                temp.pop_back();
            
        
    
    vector<vector<int>> permute(vector<int>& nums) 
        vector<bool> traverse(nums.size(), false);
        backtrack(nums, traverse);
        return ans;
    
;

类里temp是装一种排列的数组,ans则是全排列所有答案的二维数组。
主要是要熟练backtrack的操作:
回溯总是需要有一个"回去"的条件,也就是终止递归的条件,否则就是无穷无尽的递归直到"爆栈"。此处我们设置的终止条件就是temp装满了一次排列,也就是size和原数组相等时就找到一次排列了。
而对于如何寻找排列主要还是看循环里干了啥事。traverse是我们的一个访问数组,对应于下标而不是元素本身,没访问前所有下标都是false的。nums数组是题目给出的,希望我们找到所有排列的数组。
如果某个元素没被访问过就可以装入temp,且修改标记。当此元素后面的所有位置完成全排列后就会回退到这个元素,此时我们将其从当前位置弹出并修改访问位即可。
按照上述步骤就可以得到简单全排列。

需要剪枝的全排列

需要剪枝的最大原因就是数组中出现重复数字了。

如果不带剪枝,那么每次判定到终止条件时我们还需要嵌套一层循环去判定这个temp是否已经在ans里出现过,没出现过才加入此种排列方式。
这样的效率极其低下,因此要学会如何在递归过程中剪枝而不是在递归结果处判定。

朴素回溯的逻辑在于,比如有例一数组[1, 1, 2]:
则将下标为0的元素置首,剩下的全排列
再将下标为1的元素置首,剩下的全排列
最后下标为2的元素置首,剩下的全排列

对于一个位置loc,假设现在企图把nums数组中下标为index的数字填到loc处,那么index之前的那些数字都已经被尝试过填到loc处。
显然,如果当前index这个数字与index - 1下标处的数字相同的话,整个排列就重复了!所以假如相同这个情况,loc处就不需要再填index下标的数字了,换下一个!
代码如下:

class Solution 
public:
    vector<vector<int>> ans;
    vector<int> vec;

    void dfs(vector<bool>& visited, vector<int>& nums)
        if(vec.size() == nums.size())
            ans.push_back(vec);
            return;
        
        for(int i = 0 ; i < nums.size() ; i ++)
            if(i - 1 >= 0 && !visited[i - 1] && nums[i] == nums[i - 1])
                continue;
            if(!visited[i])
                visited[i] = true;
                vec.push_back(nums[i]);
                dfs(visited, nums);
                vec.pop_back();
                visited[i] = false;
            
        
    

    vector<vector<int>> permuteUnique(vector<int>& nums) 
        sort(nums.begin(), nums.end(), [](int item1, int item2)
            return item1 > item2;
        );
        int sz = nums.size();
        vector<bool> visited(sz, false);
        dfs(visited, nums);
        return ans;
    
;

还有一个关键点就是nums数组得事先排序。

以上是关于基于剪枝的回溯法的主要内容,如果未能解决你的问题,请参考以下文章

基于剪枝的回溯法

0-1背包问题的回溯法中,剪枝用的上界函数问题

HDU - 1010 (DFS回溯法 + 奇偶剪枝)

回溯法

回溯算法总结

解数独算法的实现——剪枝优化