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

lisanndesu2025-01-25 12:52

dfs解决全排列&子集

1.全排列

全局变量+回溯

code

复制代码

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

    void dfs(vector<int>& 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<vector<int>> ans;
    vector<int> cur;
    vector<vector<int>> subsets(vector<int>& nums) {
        dfs(nums, 0);
        return ans;
    }

    void dfs(vector<int>& 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);
    }
};