1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54
| class Solution { public: vector<vector<string>> ans; bool isvalid(int row,int col,vector<string>& chessboard,int n) { int count = 0;
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; --j,--i ) { if(chessboard[i][j] == 'Q') return false; }
for(int i = row-1, j = col+1; j < n && i>=0; ++j,--i) { if(chessboard[i][j] == 'Q') return false; } return true; }
void backtracking(int n , int row,vector<string>& chessboard) { if(row == n) { ans.push_back(chessboard); return; } for(int i = 0; i < n;++i) { if(isvalid(row,i,chessboard,n)) { chessboard[row][i] = 'Q'; backtracking(n,row+1,chessboard); chessboard[row][i] = '.'; } } } vector<vector<string>> solveNQueens(int n) { vector<string> chessboard(n,string(n,'.')); backtracking(n,0,chessboard); return ans; } };
|