N皇后问题
在N*N的方格棋盘放置了N个皇后,使得它们不相互攻击(即任意2个皇后不允许处在同一排,同一列,也不允许处在与棋盘边框成45角的斜线上。
你的任务是,对于给定的N,求出有多少种合法的放置方法。
Input
共有若干行,每行一个正整数N≤10,表示棋盘和皇后的数量;如果N=0,表示结束。
Output
共有若干行,每行一个正整数,表示对应输入行的皇后的不同放置数量。
Sample Input
1 8 5 0
Sample Output
1 92 10
思路:
DFS回溯
代码:
#include<iostream>
#include<algorithm>
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define dep(i,a,b) for(int i=a;i>=b;i--)
#define ll long long
#include<string.h>
#include<queue>
#include<stack>
#include<cstdio>
using namespace std;
const int maxn=55+6;
#define mod 1000000007
#define INF 0x3f3f3f3f
int a[maxn][maxn];//n皇后问题
bool vis[maxn][maxn];
int n;
int ans;
bool check(int r,int c)
{
rep(i,1,n)
{
rep(j,1,n)
{
if(!vis[i][j])
continue;
if(j==c)
return 0;
if(abs(r-i)==abs(c-j))
return 0;
}
}
return 1;
}
void dfs(int r)
{
//cout<<r<<endl;
if(r==n+1)
{
ans++;
return ;
}
rep(i,1,n)
{
if(check(r,i))//考虑当前
{
vis[r][i]=1;
dfs(r+1);
vis[r][i]=0;
}
}
}
int ok[maxn];
int main()
{
while(cin>>n&&n)
{
if(ok[n])
{
cout<<ok[n]<<endl;
continue;
}
memset(vis,0,sizeof(vis));
ans=0;
dfs(1);
ok[n]=ans;
cout<<ans<<endl;
}
}
2n皇后问题
时间限制: 1Sec 内存限制: 128MB 提交: 658 解决: 324
题目描述
给定一个n*n的棋盘,棋盘中有一些位置不能放皇后。现在要向棋盘中放入n个黑皇后和n个白皇后,使任意的两个黑皇后都不在同一行、同一列或同一条对角线上,任意的两个白皇后都不在同一行、同一列或同一条对角线上。问总共有多少种放法?n小于等于8。
输入
输入的第一行为一个整数n,表示棋盘的大小。
接下来n行,每行n个0或1的整数,如果一个整数为1,表示对应的位置可以放皇后,如果一个整数为0,表示对应的位置不可以放皇后。
输出
输出一个整数,表示总共有多少种放法。
样例输入
4 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1
样例输出
2
#include<iostream>
#include<algorithm>
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define dep(i,a,b) for(int i=a;i>=b;i--)
#define ll long long
#include<string.h>
#include<queue>
#include<stack>
#include<cstdio>
using namespace std;
const int maxn=5+6;
#define mod 1000000007
#define INF 0x3f3f3f3f
int a[maxn][maxn];//2n皇后问题
bool v1[maxn][maxn],v2[maxn][maxn];
int ans;
int n;
bool checkho(int r,int c)
{
if(v2[r][c])
return 0;
rep(i,1,n)
{
rep(j,1,n)
{
if(v1[i][j])
{
if(c==j||abs(r-i)==abs(j-c))
return 0;//不合法
}
}
}
return 1;
}
bool checkhi(int r,int c)
{
if(v1[r][c])
return 0;
rep(i,1,n)
{
rep(j,1,n)
{
if(v2[i][j])
{
if(c==j||abs(r-i)==abs(j-c))
return 0;//不合法
}
}
}
return 1;
}
void dfs(int r)
{
//cout<<r<<endl;
if(r==n+1)
{
ans++;
return ;
}
rep(i,1,n)
{
//先放红、再放黑
if(checkho(r,i)&&a[r][i])
{
v1[r][i]=1;
rep(j,1,n)
{
if(checkhi(r,j)&&!v1[r][j]&&a[r][j])
{
v2[r][j]=1;
dfs(r+1);
v2[r][j]=0;
}
}
v1[r][i]=0;
}
}
}
int main()
{
cin>>n;
rep(i,1,n)
{
rep(j,1,n)
{
cin>>a[i][j];
}
}
dfs(1);
cout<<ans<<endl;