代码随想录算法训练营第三十 |332.重新安排行程 、 51. N皇后、 37. 解数独

今天三道题难度比较大，使用回溯方法的思路挺清晰的，我在一些数据存储和映射关系表示上有点薄弱，手写的笔记思路有些乱，代码中的注解可以补充着看。

332.重新安排行程

https://programmercarl.com/0332.%E9%87%8D%E6%96%B0%E5%AE%89%E6%8E%92%E8%A1%8C%E7%A8%8B.html

class Solution {
public:
    //          <出发机场， map<到达机场，航班次数>>
    unordered_map<string, map<string, int>> targets;
    //找到一条合适的行程安排就可以结束，使用的bool类型
    bool backtracking(int ticketsNum, vector<string>& result) {
        if(result.size() == ticketsNum+1) {
            return true;
        }
        //targets[result[result.size()-1]]中获取本次for循环处理的元素集合
        //target相当于i
        for(pair<const string, int>&target : targets[result[result.size()-1]]) {
            if(target.second > 0) {
                result.push_back(target.first);
                target.second--;
                if(backtracking(ticketsNum,result)) return true;
                result.pop_back();
                target.second++;
            }
        }
        return false;
    }
    vector<string> findItinerary(vector<vector<string>>& tickets) {
        //这里的result就相当于前面做题用的path，只需要记录一条合适的路线
        vector<string> result;
        for(const vector<string>& vec : tickets) { //vec是取tickets中的第一个from to
            targets[vec[0]][vec[1]]++;  //更新 [vec[0]]作为from [vec[1]]作为to的映射关系
        }
        result.push_back("JFK");  //所有都是从JKF机场出发
        backtracking(tickets.size(),result);
        return result;
    }
};

51. N皇后

https://programmercarl.com/0051.N%E7%9A%87%E5%90%8E.html
视频讲解：https://www.bilibili.com/video/BV1Rd4y1c7Bq

class Solution {
public:
    vector<vector<string>> result;
    void backtracking(int n,int row, vector<string>& chessboard) {
        if(row==n) {
            result.push_back(chessboard);
            return;
        }
        // for循环处理的是一层的固定row，给第row行的每一列放置Q查看效果
        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;
        }
        //检查45度方向是否有皇后
        for(int i=row-1,j=col-1;i>=0&&j>=0;i--,j--) {
            if(chessboard[i][j]=='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;
        }
        return true;
    }
    vector<vector<string>> solveNQueens(int n) {
        result.clear();
        std::vector<std::string> chessboard(n,std::string(n,'.'));
        backtracking(n,0,chessboard);
        return result;
    }
};

37. 解数独

https://programmercarl.com/0037.%E8%A7%A3%E6%95%B0%E7%8B%AC.html
视频讲解：https://www.bilibili.com/video/BV1TW4y1471V

class Solution {
public:

    bool backtracking(vector<vector<char>>& board) {
        //遍历行
        for(int i=0;i<board.size();i++) {
            //遍历列  这里的j的上限影响大吗？board[0].size()
            for(int j=0;j<board[0].size();j++) {
                //找到可以填的地方
                if(board[i][j]=='.') {
                    //从1-9字符还是尝试
                    for(char k = '1'; k<='9';k++) {
                        //如果合法就把k填入，递归然后回溯
                        if(isValid(i,j,k,board)) {
                            board[i][j]=k;
                            if(backtracking(board)) {
                                return true;
                            }
                            board[i][j]='.';
                        }
                    }
                    //都不行，就返回false;
                    return false;
                }
            }
        }
        return true;
    }
    bool isValid(int row,int col,char k,vector<vector<char>>& board) {
        //判断行是否重复
        for(int i=0;i<9;i++) {
            if(board[row][i] == k) {
                return false;
            }
        }
        //判断列是否重复
        for(int j=0;j<9;j++) {
            if(board[j][col]== k)
                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]==k)
                    return false;
            }
        }
        return true;
    }
    void solveSudoku(vector<vector<char>>& board) {
        backtracking(board);
    }
};