刚开始怎么也想不通怎么斜向判断,看了详解,服了。。戳这里N皇后详解
准确的说他是用了滚动数组,分别用三行记录正对角线,纵向,反对角线方向所放置过皇后所能影响的状态记录。若在攻击范围内,则标记1。而且越想越觉得精妙,数组是动态的,dfs遍历每行,for遍历每列,每个状态的判断都在前一个判断的基础上,而且加之回溯节省时间,再加上打表的小技巧,这样的程序真的难得一见哪。。。
ps:开学感觉浪费好多时间,如果这些能利用起来就好了
#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <queue>
#include <iostream>
using namespace std;
const int N = 30;
const int INF = 1000000;
int vis[3][N], ans[N];
int n, sum;
void dfs(int row)
{
int i;
if(row == n + 1)
{
sum ++;
return;
}
for(i = 1; i <= n; i ++)
{
if(vis[0][row - i + n] == 0 && vis[1][i] == 0 && vis[2][i + row] == 0)
{
vis[0][row - i + n] = vis[1][i] = vis[2][i + row] = 1;
dfs(row + 1);
vis[0][row - i + n] = vis[1][i] = vis[2][i + row] = 0;//回溯
}
}
}
int main()
{
// freopen("in.txt", "r", stdin);
for(n = 1; n <= 10; n ++) //注意这里是n而不能为i因为n必须同步
{
memset(vis, 0, sizeof(vis));
sum = 0;
dfs(1);
ans[n] = sum;
}
while(~scanf("%d", &n) && n)
printf("%d\n", ans[n]);
return 0;
}