【代码随想录算法训练Day29】LeetCode 491.非递减子序列、LeetCode 46.全排列、LeetCode 47.全排列II

Frostnova丶2024-06-10 22:47

Day29 回溯第五天

LeetCode 491.非递减子序列

这道题比起子集II来说又多了要考虑的点，首先我们不能排序来去重了，因为这样会增加很多不合题意的递增序列。所以我们使用set来去重。同时还要注意题目对答案子集要求内部元素的数量要大于1。

这个代码有一个精髓：set是在回溯内部建立的，也就是说他会每次遍历都创建一个新的set，自动完成全新回溯操作的去重工作，所以不需要专门在递归后对set进行回溯了。

cpp 复制代码

class Solution {
public:
    vector<int> path;
    vector<vector<int>> res;

    void backtrack(vector<int>& nums,int startIndex){
        if(path.size()>1) res.push_back(path);
        unordered_set<int> uset;
        for(int i=startIndex;i<nums.size();i++){
            if((!path.empty() && nums[i]<path.back()) || uset.find(nums[i])!=uset.end())
                continue;
            uset.insert(nums[i]);
            path.push_back(nums[i]);
            backtrack(nums,i+1);
            path.pop_back();
        }
    }

    vector<vector<int>> findSubsequences(vector<int>& nums) {
        backtrack(nums,0);
        return res;
    }
};

LeetCode 46.全排列

全排列比起前面的就简单一些了，我们甚至不需要向后遍历，只需要用一个数组确认是否遍历过，然后不断从头开始递归即可。

cpp 复制代码

class Solution {
public:
    vector<int> path;
    vector<vector<int>> res;

    void backtrack(vector<int>& nums,vector<bool>& used){
        if(path.size()==nums.size()){
            res.push_back(path);
            return;
        } 
        
        for(int i=0;i<nums.size();i++){
            if(used[i]==true) continue;
            used[i]=true;
            path.push_back(nums[i]);
            backtrack(nums,used);
            path.pop_back();
            used[i]=false;
        }
    }

    vector<vector<int>> permute(vector<int>& nums) {
        vector<bool> used(nums.size(),false);
        backtrack(nums,used);
        return res;
    }
};

LeetCode 47.全排列II

与前一题相比，增加了一个去重的过程，这也是之前的题目里出现过的。

cpp 复制代码

class Solution {
public:
    vector<int> path;
    vector<vector<int>> res;

    void backtrack(vector<int>& nums,vector<bool>& used){
        if(path.size()==nums.size()){
            res.push_back(path);
            return;
        }
        for(int i=0;i<nums.size();i++){
            if(i>0 && nums[i]==nums[i-1] && used[i-1]==false)
                continue;
            if(used[i]==false){
                used[i]=true;
                path.push_back(nums[i]);
                backtrack(nums,used);
                path.pop_back();
                used[i]=false;
            }
        }
    }

    vector<vector<int>> permuteUnique(vector<int>& nums) {
        sort(nums.begin(),nums.end());
        vector<bool> used(nums.size(),false);
        backtrack(nums,used);
        return res;
    }
};

今天是排列，加油！