51. N 皇后
我们已知一行只能有一个皇后,所以可以用一维数组chessboard表示皇后的位置,即(i, chessboard[i])表示第i个皇后的位置
通过上图全排列,剪枝的方式可以获得满足规则的皇后位置,验证是否冲突时,我们只需要验证和前面的皇后是否冲突即可,因为我们是一行一行放置皇后
class Solution {
public:
vector<int> chessboard;
vector<vector<string>> ans;
bool isCollide(int pos){
for(int i = 0; i < pos; i++){
if(chessboard[i] == chessboard[pos] || abs(i - pos) == abs(chessboard[i] - chessboard[pos])){
return true;
}
}
return false;
}
void func(int startIndex, int n){
if(startIndex == n){
vector<string> tmp;
for(int i = 0; i < n; i++){
string path(n, '.');
path[chessboard[i]] = 'Q';
tmp.push_back(path);
}
ans.push_back(tmp);
}
for(int i = startIndex; i < n; i++){
// 这里开始一行一行放皇后,一行放一个
swap(chessboard[startIndex], chessboard[i]);
if(!isCollide(startIndex)){
// 当前摆放的位置(startIndex, chessboard[startIndex])是否和前面的皇后冲突
func(startIndex + 1, n);
}
swap(chessboard[startIndex], chessboard[i]);
}
}
vector<vector<string>> solveNQueens(int n) {
for(int i = 0; i < n; i++){
chessboard.push_back(i);
}
func(0, n);
return ans;
}
};
37. 解数独
暴力法填写即可
class Solution {
public:
// 检查在board[row][col]的位置放val是否合法
bool isValid(int row, int col, char val, const vector<vector<char>>& board){
// 检查列
for(int i = 0; i < 9; i++){
if(board[i][col] == val){
return false;
}
}
// 检查行
for(int j = 0; j < 9; j++){
if(board[row][j] == val){
return false;
}
}
// 检查board[row][col]所在的3x3宫格
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;
}
bool func(vector<vector<char>>& board){
// 这里从0开始...
for(int i = 0; i < 9; i++){
for(int j = 0; j < 9; j++){
if(board[i][j] != '.'){
// board[i][j]已经被填写了
continue;
}
// 开始在board[i][j]位置尝试放置数字
for(char val = '1'; val <= '9'; val++){
if(!isValid(i, j, val, board)){
continue;
}
board[i][j] = val;
if(func(board)){
// 找到合法的解,返回即可,不用再回溯了
return true;
}
board[i][j] = '.';
}
// 9个数字都不能放在board[i][j]位置,当前分支失败,需要向上回溯
return false;
}
}
return true;
}
void solveSudoku(vector<vector<char>>& board) {
func(board);
}
};