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

cpp 复制代码

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) {
        targets.clear();
        vector<string> result;
        for(const vector<string>& vec:tickets)
        {
            targets[vec[0]][vec[1]]++;
        }
        result.push_back("JFK");
        backtracking(tickets.size(),result);
        return result;
    }
};

51.N皇后

思路：这道题做了之前的回溯算法基本上思路理解起来还是很容易的

cpp 复制代码

class Solution {
public:
    vector<vector<string>> result;
    //isvalid函数的作用：检查每一行哪个位置能放呢？
    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;
            }
        }
        //检查135度的情况，检查放的这个位置的前一行，右边一列的位置放置'Q'没有
        for(int i=row-1,j=col+1;i>=0 && j<n;i--,j++)
        {
            if(chessboard[i][j]=='Q')
            {
                return false;
            }
        }
        return true;
    }
    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]='.';
            }
        }
    }
    vector<vector<string>> solveNQueens(int n) {
        result.clear();
        vector<string> chessboard(n,string(n,'.'));//创建一个二维数组，相当于一个棋盘！
        backtracking(n,0,chessboard);
        return result;
    }
};

37.解数独

cpp 复制代码

class Solution {
public:
    bool isvalid(int row,int col,char val,vector<vector<char>>& board)
    {
        for(int i=0;i<9;i++)
        {
            if(board[row][i] == val)
                return false;
        }
        for(int i=0;i<9;i++)
        {
            if(board[i][col]==val)
                return false;
        }
        int startRow=(row/3)*3;//这里处理的目的，回到每个3x3格子的起点
        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;
    }
    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]!='.') continue;
                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;
    }
    void solveSudoku(vector<vector<char>>& board) {
        backtracking(board);
    }
};