皇后(queen)

时间:2023-03-09 15:41:34
皇后(queen)

皇后(queen)
【题目描述】
  众所不知,rly现在不会玩国际象棋。但是,作为一个OIer,rly当然做过八皇后问题。这里再啰嗦几句,皇后可以攻击到同行同列同对角线,在n*n的方格中摆n个皇后使其互不攻击到,求不同的解的数量,这就是经典的n皇后问题。现在问题推广到n皇后问题,这个问题对于你而言实在是小菜一叠。但因为上一次rly把棋盘弄破了,又拿不出新的,所以rly打算难一点点,问题就是破棋盘上的n皇后问题。他想知道……(你们懂的)。
  棋子都是相同的。
【输入说明】
  一行,一个正整数N。
  接下来N行,每行N个数,要么为0,表示没坏,要么1,表示坏了。
【输出说明】
  一行,输出不同的解的数量。
【样例输入】
  4
  1 0 1 1
  1 1 1 0
  0 1 1 1
  1 1 0 1
【样例输出】
  1
【数据范围】
  对于40%的数据,N<=13。
  对于100%的数据,N<=16。
  其中有30%的数据,棋盘没有破(你可以认为rly又去买了一个新的)。

【题目分析】

  打表+....唉,几乎照搬了N皇后问题的代码,做了一点小小的处理,答案是可以跑出来的,但是有两个点严重TLE

#include <iostream>
#include <cstdio>
using namespace std;
int n,sum,a[];
bool b[],c[],d[];
int mapp[][];
int ans[]={,,,,,,,,,,,,,,,,};
void search2(int i)
{
for(int j=;j<=n;j++)
if((!b[j])&&(!c[i+j])&&(!d[i-j+n-])&&(!mapp[i][j]))
{
a[i]=j;
b[j]=;
c[i+j]=;
d[i-j+n-]=;
if(i==n) sum++;
else search2(i+);
b[j]=;
c[i+j]=;
d[i-j+n-]=;
}
}
int main()
{
freopen("queen.in","r",stdin);
freopen("queen.out","w",stdout);
int flag=;
scanf("%d",&n);
for(int i=;i<=n;i++)
for(int j=;j<=n;j++)
{
scanf("%d",&mapp[i][j]);
if(mapp[i][j]==)
flag=;
}
if(flag==)
cout<<ans[n];
else
{
search2();
cout<<sum;
}
fclose(stdin);fclose(stdout);
return ;
}

考场80