2n皇后 - 回溯

时间:2023-03-10 00:27:21
2n皇后 -  回溯

题目地址:http://www.51cpc.com/web/problem.php?id=1172

Summarize:

1. 递归回溯;

2. 先扫完一种皇后,再扫描另一种;

3. 循环输入;

4. 每列每种皇后必有一个,依次搜索;

附代码:

/*
2018-07-24
2n皇后 -回溯 每列必有一黑一白两个皇后,依次搜索;
将其中一种皇后放置完后,放置第二种皇后;
*/
#include<iostream>
#include<cmath>
using namespace std; #define LL long long
const int N = +;
int n, vis[N][N];
LL ans=; bool check(int x, int y, int queen)
{
if(vis[x][y] != ) return false;
if(x< || x>=n || y< || y>=n) return false;
for(int i=; i<n; i++) {
if(vis[x][i] == queen || vis[i][y] == queen)
return false;
if(vis[x+i][y+i] == queen && x+i<n && y+i<n || vis[x-i][y-i] == queen && x>=i && y>=i
|| vis[x+i][y-i] == queen && y>=i && x+i<n || vis[x-i][y+i] == queen && x>=i && y+i<n)
return false;
}
return true;
} void bqueen(int x) //3 - black
{
if(x == n) {
ans++;
return;
} for(int i=; i<n; i++)
{
if(!check(x, i, )) continue;
vis[x][i] = ;
bqueen(x+);
vis[x][i] = ;
}
} void wqueen(int x) //2 - white
{
if(x == n) {
bqueen();
return;
} for(int i=; i<n; i++)
{
if(!check(x, i, )) continue;
vis[x][i] = ;
wqueen(x+);
vis[x][i] = ;
}
} int main()
{
ios::sync_with_stdio(false); while(cin>>n)
{
for(int i=; i<n; i++)
for(int j=; j<n; j++)
cin>>vis[i][j];
ans=;
if(n) wqueen();
cout<<ans<<endl;
} return ;
}