【BZOJ】1087【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

$Solution$

大暴搜……压位好写……
预处理可行状态，连边，然后$DP$

#include<stdio.h>

long long f[10][82][512],ans;

int n,k,tot,sum[512],poi[512],st;
bool map[512][512];

void dfs(const int &p,const int &mask,const int &t)
{
    if (p==n)
    {
        poi[tot]=mask;sum[tot++]=t;
        if (mask==0) st=tot-1;
        return;
    }
    if ((mask & (1<<(p-1)))==0) dfs(p+1,mask | (1<<p),t+1);
    dfs(p+1,mask,t);
}

inline void build()
{
    for (int i=0;i<tot;i++) for (int j=0;j<tot;j++) 
     if ((poi[i] & poi[j])==0 && ((poi[i]<<1) & poi[j])==0 && ((poi[i]>>1) & poi[j])==0) map[i][j]=map[j][i]=1;
}

int main()
{
    scanf("%d%d",&n,&k);
    dfs(1,0,0);dfs(1,1,1);
    build();
    f[0][0][st]=1;
    for (int i=1;i<=n;i++) for (int j=0;j<=k;j++) 
     for (int o=0;o<tot;o++) if (sum[o]<=j) for (int p=0;p<tot;p++) if (map[o][p])
      f[i][j][o]+=f[i-1][j-sum[o]][p];
    for (int i=0;i<tot;i++) ans+=f[n][k][i];
    printf("%lld",ans);
}