二刷代码随想录训练营Day 25|491.递增子序列、46.全排列、47.全排列 II、332.重新安排行程、51.n皇后、37.解数独

1.递增子序列

代码:

class Solution {
private:
    vector<int> path;
    vector<vector<int>> result;
    void backtracking(vector<int> &nums,int startIndex){
        if(path.size() > 1){
            result.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();
        }
    }
public:
    vector<vector<int>> findSubsequences(vector<int>& nums) {
        backtracking(nums,0);
        return result;
    }
};

 note:这里不用used数组来去重的原因是,本题不能对数组进行排序,会破坏原数组元素间的相对关系。

2.全排列

代码:

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

 note:这里是排列问题,因此我们不用startIndex。

但是要进行树枝去重,也就是用过的元素不能再用,这里的used数组在使用时,因为不去考虑有重复元素的情况,因此使用前不需要对数组进行排序,也不用比较和之前元素是否相等。

3.全排列2

代码:

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

 note:这里要进行树层去重,对数组先进行排序,然后used数组树层去重的逻辑和之前的一样。

 4.重新安排行程 

代码随想录

代码:

class Solution {
private:
    unordered_map<string,map<string,int>> targets; // <出发机场,<到达机场,航班次数>>
    bool backtracking(int ticketNum,vector<string>& result){
        if(result.size() == ticketNum + 1){
            return true;
        }
        for(pair<const string,int>& target : targets[result[result.size() - 1]]){
            if(target.second > 0){ // 记录到达机场是否飞过了
                result.push_back(target.first);
                target.second--;
                if(backtracking(ticketNum,result)) return true;
                result.pop_back();
                target.second++;
            }
        }
        return false;
    }
public:
    vector<string> findItinerary(vector<vector<string>>& tickets) {
        vector<string> result;
        for(const vector<string>& vec:tickets){
            targets[vec[0]][vec[1]]++; // 记录映射关系
        }
        result.push_back("JFK"); // 起始机场
        backtracking(tickets.size(),result);
        return result;
    }
};

 note:处于一个可以看懂,但是感觉自己肯定写不出来的状态。

5.n皇后

代码:

class Solution {
private:
    vector<vector<string>> result;
    void backtracking(int n, int row, vector<string>& chessboard){
        if(row == n){
            result.push_back(chessboard);
            return;
        }
        for(int col = 0; col < n; col++){
            if(isValid(row,col,chessboard,n)){
                chessboard[row][col] = 'Q';
                backtracking(n,row + 1,chessboard);
                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;
            }
        }
        // 检查135
        for(int i = row - 1,j = col + 1; i >= 0&& j < n; i--,j++){
            if(chessboard[i][j] == 'Q'){
                return false;
            }
        }
        // 检查45
        for(int i = row - 1,j = col - 1; i >= 0 && j >= 0; i--,j--){
            if(chessboard[i][j] == 'Q'){
                return false;
            }
        }
        return true;
    }
public:
    vector<vector<string>> solveNQueens(int n) {
        vector<string> chessboard(n,string(n,'.'));
        backtracking(n,0,chessboard);
        return result;
    }
};

note:我感觉这道题是,回溯函数的参数比较难想,以及判断45和135°对角线比较难写(最好画个图看看)

6.解数独

代码:

class Solution {
private:
    bool backtracking(vector<vector<char>>& board){
        for(int i = 0; i < board.size(); i++){
            for(int j = 0; j < board[0].size(); j++){
                if(board[i][j] == '.'){
                    for(char k = '1'; k <= '9'; k++){
                        if(isValid(i,j,k,board)){
                            board[i][j] = k;
                            if(backtracking(board)) return true;
                            board[i][j] = '.';
                        }
                    }
                    return false;
                }
            }
        }
        return true;
    }
    bool isValid(int row,int col,char val,vector<vector<char>> &board){
        for(int i = 0; i < 9; i++){
            if(board[i][col] == val || board[row][i] == val){
                return false;
            }
        }
        int startRow = (row / 3) * 3;
        int startCol = (col / 3) * 3;
        for(int i = startRow; i < startRow + 3; i++){
            for(int j = startCol; j < startCol + 3; j++){
                if(board[i][j] == val){
                    return false;
                }
            }
        }
        return true;
    }
public:
    void solveSudoku(vector<vector<char>>& board) {
        backtracking(board);
    }
};

 

  • 1
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值