bzoj1087: [SCOI2005]互不侵犯King

Description

　　在N×N的棋盘里面放K个国王，使他们互不攻击，共有多少种摆放方案。国王能攻击到它上下左右，以及左上
左下右上右下八个方向上附近的各一个格子，共8个格子。

Input

　　只有一行，包含两个数N，K （ 1 <=N <=9, 0 <= K <= N * N）

Output

　　方案数。

Sample Input

3 2
Sample Output

16

题解

经典的状态压缩DP。
因为不能攻击左上右上正上方的，所以要记录上一行的状态。每一行的状态用2进制记录，0表示没国王，1表示有国王，这样配合位运算就可以进行状态转移了。

#include<iostream>
#include<cstdio>

using namespace std;

int n,m,all,cnt[512];

long long f[10][100][512];

bool c1[512],c2[512][512];

long long ans;

void pre()
{
     int s;
     for(int i=0;i<=all;i++)
        if((i&(i>>1))==0)
           {
           s=0;
           for(int x=i;x;x>>=1)s+=(x&1);
           cnt[i]=s;c1[i]=1;
           }
     for(int i=0;i<=all;i++)if(c1[i])
        for(int j=0;j<=all;j++)if(c1[j])
           if(((i&j)==0)&&((i&(j>>1))==0)&&((j&(i>>1))==0))
              c2[i][j]=1;
     }
int main()
{
    scanf("%d%d",&n,&m);
    all=(1<<n)-1;
    pre();
    for(int i=0;i<=all;i++)if(c1[i])f[1][cnt[i]][i]=1;
    for(int j=1;j<n;j++)
       for(int k=0;k<=all;k++)if(c1[k])
          for(int i=0;i<=all;i++)if(c1[i])
             if(c2[k][i])
                for(int p=cnt[k];p+cnt[i]<=m;p++)
                    f[j+1][p+cnt[i]][i]+=f[j][p][k];
    long long ans=0;
    for(int i=0;i<=all;i++)ans+=f[n][m][i];
    printf("%lld",ans);
    return 0;
    }