Description
在N×N的棋盘里面放K个国王,使他们互不攻击,共有多少种摆放方案。国王能攻击到它上下左右,以及左上
左下右上右下八个方向上附近的各一个格子,共8个格子。
Input
只有一行,包含两个数N,K ( 1 <=N <=9, 0 <= K <= N * N)
Output
方案数。
Sample Input
3 2
Sample Output
16
不由得想起了八皇后问题。。。
五百行,搜索果断T掉。考虑状压,以2^9存下每一行的状态,一开始预处理便可筛掉不合理的情况。然后DP,f[i][j][k]表示到第i行在这一行选择了j个国王状态为k的方案数,记得开long long
#include<cstdio>
#include<cstring>
#include<iostream>
#include<iostream>
#ifdef WIN32
#define AUTO "%I64d"
#else
#define AUTO "%lld"
#endif
using namespace std;
const int maxn=550;
int n,m,tot,sum[maxn];
long long f[11][100][maxn];
bool line[maxn],colunm[maxn][maxn];//这里单词运用有误,其实colunm是相邻两行
void init()//筛掉本身不合法状态
{
for (int i=0;i<=tot;i++) if ((i&(i>>1))==0)
{
for (int j=i;j>0;j>>=1) if ((j&1)==1) sum[i]++;
line[i]=true;
}
for (int i=0;i<=tot;i++) if (line[i])
for (int j=0;j<=tot;j++) if (line[j])
if ((i&j)==0&&(j&(i>>1))==0&&(i&(j>>1))==0)
colunm[i][j]=true;
for (int i=0;i<=tot;i++) if(line[i]) f[1][sum[i]][i]=1;
}
int main()
{
scanf("%d%d",&n,&m);
tot=(1<<n)-1;
init();
for(int j=1;j<n;j++)
for(int k=0;k<=tot;k++) if(line[k])
for(int i=0;i<=tot;i++) if(line[i])
if(colunm[k][i])
for(int p=sum[k];p+sum[i]<=m;p++)
f[j+1][p+sum[i]][i]+=f[j][p][k];
long long ans=0;
for (int i=0;i<=tot;i++) ans+=f[n][m][i];
printf(AUTO,ans);
return 0;
}