题意:在一个棋盘中问有几种方法可以放置K个“国王”。
做法:国王可以攻击四面八方的棋子,所以必须加一个数组(fig)记录每个国王的攻击状态,由对称性,只要记录下半边的即可。之后,就是水的状态压缩了。
#include<stdio.h>
#include<string.h>
int state[150],fig[150],one[150];
long long dp[12][150][102];
int n,k,all;
void dfs(int lie,int s)
{
if(lie>=n)
{
state[all]=s;
for(int i=0;i<n;i++)
if(1<<i&s)
{
one[all]++;
if(i>1)
fig[all]|=1<<(i-1);
if(i<n-1)
fig[all]|=1<<(i+1);
fig[all]|=1<<i;
}
all++;
return;
}
dfs(lie+1,s);
dfs(lie+2,s|1<<lie);
}
int main(void)
{
int i,j,t,x;
long long ans;
while(scanf("%d%d",&n,&k)!=EOF)
{
memset(fig,0,sizeof(fig));
memset(dp,0,sizeof(dp));
memset(one,0,sizeof(one));
all=0;
dfs(0,0);
dp[0][0][0]=1;
for(i=1;i<=n;i++)
for(j=0;j<all;j++)
for(t=0;t<all;t++)
if((fig[t]&state[j])==0&&(fig[j]&state[t])==0)
for(x=0;x<=k-one[j];x++)
if(dp[i-1][t][x])
dp[i][j][x+one[j]]+=dp[i-1][t][x];
ans=0;
for(i=0;i<all;i++)
ans+=dp[n][i][k];
printf("%I64d\n",ans);
}
return 0;
}