教你做回溯题(Leetcode全排列和全排列II)-第三期

前两期指路：

第一期：第一期

第二期：第二期

例题

全排列

给定一个不含重复数字的数组，返回所有可能的全排列，对于组合集合来说，要考虑顺序

①选择一个实例[1,2,3]画图

 ②确定结束条件

由于是全排列，所以最后结果的集合一定包含所有的数，那么结束条件是path的长度与题目给的nums一样

//结束条件
if(path.size()==nums.size()){
    res.push_back(path);
}

③确定选择列表，我们发现这题选择列表相对于前几期没那么有规律。

勘误：下图中间那个选择列表应该是[1,3]

选择列表如图所示，会变，也就是说一条深度路径下，前面使用过的，后面不能再使用，比如看最左边的一条路径，开始选择列表是123，选择了1，那么1不能用了，选择列表变为了23，选择了2，那么2不能用了，选择列表变为了3。

这时候需要定义一个used记录使用情况

vector<bool> used(n,false);

选择列表为

for(int i=0;i<nums.size();i++){
    //未使用过
    if(!used[i]){
              
    }
}

④剪枝，这题不需要剪枝，确定了选择列表就不会有重复的了

⑤⑥⑦选择、递归、撤销

由于这题选择列表起点终点不变，只是需要根据used来判断是否使用过，所以递归不需要传i值

//选择
path.push_back(nums[i]);
used[i]=true;
//递归
dfs(nums,res,path,used);
//撤销
path.pop_back();
used[i]=false;

完整代码：

class Solution {
public:
    vector<vector<int>> permute(vector<int>& nums) {
        int n=nums.size();
        vector<bool> used(n,false);
        vector<vector<int>> res;
        vector<int> path;
        dfs(nums,res,path,used);
        return res;
    }

    void dfs(vector<int>& nums,vector<vector<int>>& res, vector<int>& path,vector<bool>& used){
        //结束条件
        if(path.size()==nums.size()){
            res.push_back(path);
        }
        for(int i=0;i<nums.size();i++){
            //未使用过
            if(!used[i]){
                //选择
                path.push_back(nums[i]);
                used[i]=true;
                //递归
                dfs(nums,res,path,used);
                //撤销
                path.pop_back();
                used[i]=false;
            }
        }
    }
};

全排列 II

给定一个可包含重复数字的序列（相对于上一题多了个给定的序列可以重复），返回不重复的全排列。

①选择一个实例[1,2,2]画图

 ②③和上一题一样

④剪枝，很明显图中有重复的需要剪枝

 和前几期一样的套路，先排序

//排序，为剪枝做准备
sort(nums.begin(), nums.end());

同一层重复不行需要剪枝，不同层可以重复不用剪枝（因为本身题目所给集合就可能有重复的）

if(i>0 && nums[i]==nums[i-1] && !used[i-1]){
    continue;
}

⑤⑥⑦选择、递归、撤销和上一题一样

完整代码：

class Solution {
public:
    vector<vector<int>> permuteUnique(vector<int>& nums) {
        vector<vector<int>> res;
        vector<int> path;
        int n=nums.size();
        vector<bool> used(n,false);
        //排序，为剪枝做准备
        sort(nums.begin(), nums.end());
        dfs(res,path,used,nums);
        return res;
    }

    void dfs(vector<vector<int>>& res, vector<int>& path, vector<bool>& used, vector<int>& nums){
        //结束条件
        if(path.size()==nums.size()){
            res.push_back(path);
            return;
        }
        for(int i=0;i<nums.size();i++){
            //使用未使用过的
            if(!used[i]){
                //如果与前一个重复且前一个在当前组合下未使用就需要去掉
                //因为对方可能在其他组合的当前层次下使用过会重复
                //如果使用过，就可以，是因为虽然值重复但在同一树枝不同树层下就可以
                if(i>0 && nums[i]==nums[i-1] && !used[i-1]){
                    continue;
                }
                //选择
                path.push_back(nums[i]);
                used[i]=true;
                //递归
                dfs(res,path,used,nums);
                //撤销
                path.pop_back();
                used[i]=false;
            }
        }
    }
};