专题三_穷举vs暴搜vs深搜vs回溯vs剪枝_全排列

dfs解决 全排列&子集

1.全排列

link:46. 全排列 - 力扣（LeetCode）

全局变量+回溯

code

class Solution {
public:
    vector> ans;
    vector cur;
    vector used;
    vector> permute(vector& nums) {
        // 暴力枚举
        used = vector(nums.size(), false);
        dfs(nums);
        return ans;
    }

    void dfs(vector& nums)
    {   // 出口
        if(cur.size() == nums.size())
        {
            ans.push_back(cur);
            return;
        }
        // 主体
        for(int i = 0; i < nums.size(); i++)
        {
            if(used[i]) continue; // 剪枝
            cur.push_back(nums[i]);
            used[i] = true;
            dfs(nums);
            used[i] = false;
            cur.pop_back(); // 回溯
        }
    }
};

2.子集

link:78. 子集 - 力扣（LeetCode）

code

class Solution {
public:
    vector> ans;
    vector cur;
    vector> subsets(vector& nums) {
        dfs(nums, 0);
        return ans;
    }

    void dfs(vector& nums, int idx)// idx表示这次选择取舍的下标
    {
        // 出口
        if(idx >= nums.size())
        {
            ans.push_back(cur);
            return;
        }
        // 主体
        //      要了
        cur.push_back(nums[idx]);
        dfs(nums, idx + 1);
        cur.pop_back();// 回溯
        //      不要
        dfs(nums, idx + 1);
    }
};