代码随想录Day25 | 491.递增子序列、46.全排列、47.全排列II、51.N皇后

491.递增子序列

不能对集合进行排序,顺序会变!

class Solution {
public:
    vector<vector<int>> res;
    vector<int> path;
    void backtracking(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]);
            backtracking(nums,i+1);
            path.pop_back();
        }
    }
    vector<vector<int>> findSubsequences(vector<int>& nums) {
        backtracking(nums,0);
        return res;
    }
};

491. 递增子序列1

树层去重:本层中重复的元素不可再次选取(每一层都维护一个set来记录本层选取过的元素,如果当前元素在set中已存在,则需要跳过本次循环。因为在每层都声明,只记录本层元素,因此不需要做回溯操作)

树枝剪枝:若当前元素小于path中最后一个元素,则不符合递增条件,需要跳过。

46.全排列

自己AC的代码:

class Solution {
public:
    vector<vector<int>> res;
    vector<int> path;
    void backtracking(vector<int>& nums,int k){
        if(path.size()==k){
            res.push_back(path);
            return;
        }
        for(int i=0;i<nums.size();i++){
            int num=nums[i];
            path.push_back(num);
            nums.erase(nums.begin()+i);
            backtracking(nums,k);
            nums.insert(nums.begin()+i,num);
            path.pop_back();
        }
    }
    vector<vector<int>> permute(vector<int>& nums) {
        int k=nums.size();
        backtracking(nums,k);
        return res;
    }
};

思路:下一层可以取的元素应该包含除了本层所取得元素之外的所有元素,因此每次取完元素放到path后,把对应的元素在数组里删除掉,保证下一层没有该元素但有其他未取元素,回溯时再加回来。

代码随想录:

class Solution {
public:
    vector<vector<int>> res;
    vector<int> path;
    void backtracking(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;
            path.push_back(nums[i]);
            used[i]=true;
            backtracking(nums,used);
            used[i]=false;
            path.pop_back();
        }
    }
    vector<vector<int>> permute(vector<int>& nums) {
        vector<bool> used(nums.size(),false);
        backtracking(nums,used);
        return res;
    }
};

用used数组记录使用过的元素,在回溯中如果发现used[i]==true,则说明树枝上使用过,跳过本次循环。

used数组说明:

在树层去重中,used[i-1]==false说明树层中使用过该元素,不可以再次使用;used[i-1]==true说明树枝上使用过,但本质上是下标不同数值相同的元素,因此可以重复使用

在全排列中,used数组记录树枝上使用过的元素,因此,只要used[i]==true,就说明nums[i]在这个树枝上已经使用过了,全排列每个元素只能使用一次,因此要跳过。

47.全排列II

自己AC的代码:

class Solution {
public:
    vector<vector<int>> res;
    vector<int> path;
    void backtracking(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)||(i>0&&nums[i]==nums[i-1]&&used[i-1]==false)) continue;
            path.push_back(nums[i]);
            cout<<nums[i]<<endl;
            used[i]=true;
            backtracking(nums,used);
            used[i]=false;
            path.pop_back();
        }
    }
    vector<vector<int>> permuteUnique(vector<int>& nums) {
        vector<bool> used(nums.size(),false);
        sort(nums.begin(),nums.end());
        backtracking(nums,used);
        return res;
    }
};

51.N皇后

class Solution {
public:
    vector<vector<string>> res;
    void backtracking(vector<string>& chessboard,int row,int n){
        if(row==n){
            res.push_back(chessboard);
            return;
        }
        for(int col=0;col<n;col++){
            if(isValid(row,col,chessboard,n)){
                chessboard[row][col]='Q';
                backtracking(chessboard,row+1,n);
                chessboard[row][col]='.';
            }
        }
    }
    bool isValid(int row,int col,vector<string>& chessboard,int n){
        for(int i=0;i<row;i++){
            if(chessboard[i][col]=='Q') return false;
        }
        for(int i=row-1,j=col-1;i>=0&&j>=0;i--,j--){
            if(chessboard[i][j]=='Q') return false;
        }
        for(int i=row-1,j=col+1;i>=0&&j<n;i--,j++){
            if(chessboard[i][j]=='Q') return false;
        }
        return true;
    }
    vector<vector<string>> solveNQueens(int n) {
        std::vector<std::string> chessboard(n, std::string(n, '.'));
        backtracking(chessboard,0,n);
        return res;
    }
};

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值