Description
在N×N的棋盘里面放K个国王,使他们互不攻击,共有多少种摆放方案。国王能攻击到它上下左右,以及左上左下右上右下八个方向上附近的各一个格子,共8个格子。
Input
只有一行,包含两个数N,K ( 1 <=N <=9, 0 <= K <= N * N)
Output
方案数。
Sample Input
3 2
Sample Output
16
题解
状态dp。状态总数很多,但可行状态很少。这是状压dp中一种常见的问题。记录可行状态是一种常用的优化技巧。
#include<cstdio>
#include<iostream>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<algorithm>
using namespace std;
int n,m,S,zz,s[1025],v[1025];
int map[1025][1025];
long long f[11][1100][90],ans=0;
int count(int x)
{
int ct=0;
while(x)
{if(x&1) ct++;
x=x>>1;
}
return ct;
}
bool check(int i,int j)
{
int x=s[i],y=s[j];
if(x&y) return false;
if(x&(y>>1)) return false;
if(x&(y<<1)) return false;
return true;
}
bool judge(int x)
{
if(x&(x<<1)) return false;
if(x&(x>>1)) return false;
return true;
}
void pre()
{
S=(1<<n)-1;
int i,j;
for(i=0;i<=S;i++)
{if(judge(i))
{s[++zz]=i; v[zz]=count(i);}
}
for(i=1;i<=zz;i++)
for(j=1;j<=zz;j++)
{if(check(i,j)) map[i][j]=map[j][i]=1;}
}
void dp()
{
int i,j,k,l;
for(i=1;i<=zz;i++) f[1][i][v[i]]=1;
for(i=1;i<n;i++)
for(j=1;j<=zz;j++)
for(k=0;k<=m;k++)
{if(f[i][j][k])
{for(l=1;l<=zz;l++)
{if(map[j][l]&&k+v[l]<=m)
f[i+1][l][k+v[l]]+=f[i][j][k];
}
}
}
for(i=1;i<=zz;i++)
ans+=f[n][i][m];
printf("%lld\n",ans);
}
int main()
{
scanf("%d%d",&n,&m);
pre(); dp();
return 0;
}