例题三、互不侵犯(四川省选2005)
题目大意:
在N×N的棋盘里面放K个国王,使他们互不攻击,共有多少种摆放方案。国王能攻击到它上下左右,以及左上左下右上右下八个方向上附近的各一个格子,共8个格子。
输入文件:
只有一行,包含两个数N,K ( 1 <=N <=9, 0 <= K <= N * N)
输出文件:
方案数。
Sample Input
3 2
Sample Output
16
每一行它对下一行的影响就体现在这一行的摆放方式以及之前总共之前放了多少个国王。所以我们可以把摆放方式作为状态,这样很容易便得到了一个粗略的方程:
F[I,J,S]=∑ F[I-1,K,T]
J,K分别表示一种摆放方式,F[I,J]表示第I行用J的摆放方式,且J与K相兼容。并且S等于T加上J这种方式在这一行放置的国王数。
首先枚举行数,枚举上一行放置的状态和这一行放置的状态,判断是否合法,然后统计这一行状态当中1的个数(即为在当前状态下这一行放国王的个数,记为T),然后枚举p=T->国王数,由上一行p-T转移,最后答案记为最后一行放国王数的状态的总和,记得开long long
废话不说代码如下:
#include<cstdio>
#include<iostream>
using namespace std;
int n,i,k,m,j,l,p;
long long f[2][82][512],ans;
inline int lowbit(const int &n)
{
int i,tot=0;
for(i=n;i;i-=(i&-i),++tot);
return tot;
}
int main()
{
scanf("%d%d",&n,&m);
int sb=(1<<n)-1;
f[0][0][0]=1;
int now=0,last=1;
for(i=1;i<=n;i++){
swap(now,last);
if (i!=1){
for(j=0;j<=sb;j++)
for(k=0;k<=m;k++)
f[now][k][j]=0;
}
for(j=0;j<=sb;j++)
if (!(j&(j<<1)))
for(k=0;k<=sb;k++)
if (!(k&(k<<1)))
{
if (!(k&j)&&!(j&(k>>1))&&!(j&(k<<1)))
{
int t=lowbit(k);
for(p=t;p<=m;p++)
f[now][p][k]+=f[last][p-t][j];
}
}
}
for(i=0;i<=sb;i++)
ans+=f[now][m][i];
cout<<ans<<endl;
}
我把数组滚动了虽然没必要