题目:https://oj.leetcode.com/problems/n-queens/
题意:N皇后问题,后边还有一个II,结题思路和这道题一样,就是输出有多少种结果,不用输出结果是什么了,更加简答了
思路:回溯,用深度优先进行递归
1.递归按照行数走,当行数>n,则表示可以打印结果了
2.每次递归都进行一个循环来试每个列可以进行
3.如果在循环中,一个列可以了,那么再写一个循环判断斜对角可不可以,判断斜对角时,比如当前递归是在第r行,那么,这个小循环的终止条件就是j<=r,再往下的行都是还没插入过皇后的,肯定不存在冲突
代码:
class Solution {
public:
vector<vector<string> > result;
int row[501]; //记录每一行的皇后放在哪个位置上,用来在判断斜线位置上有没有皇后,并用来最后把统计好的位置输出
int col[501]; //记录第几列有皇后
void dfs(int r,int n){ //递归完成第r行的情况
int i,j;
if(r == n){
vector<string> str;
for( i = 0;i<n;i++){
string tmp(n,'.');
tmp[row[i]] = 'Q';
str.push_back(tmp);
}
result.push_back(str);
return;
}
for( i = 0 ; i < n ; i++){ //大循环用来判断第i列的情况
if(col[i] == 0){
for(j = 0 ; j < r ; j++){ //小循环判断位置在[r,i]的这个格子在斜线位置上有没有冲突
if(abs(i- row[j]) == abs(j - r)) break;
}
if(j == r)
{
row[r] = i;
col[i] = 1;
dfs(r+1,n);
col[i] = 0;
row[r] = 0;
}
}
}
}
vector<vector<string> > solveNQueens(int n) {
result.clear();
dfs(0,n);
return result;
}
};
这题以前学c++的时候,简直是神题,和汉诺塔一样,都是根本看不懂的题,想想以前真好