SGU 223 Little Kings

题意:在一个棋盘中问有几种方法可以放置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;
}


你可能感兴趣的:(SGU 223 Little Kings)