bzoj 1087 互不侵犯King

题目大意：

在N×N的棋盘里面放K个国王，使他们互不攻击，共有多少种摆放方案

国王能攻击到它上下左右，以及左上左下右上右下八个方向上附近的各一个格子，共8个格子

思路：

状压dp

三维分别是当前填了几行 填的最后一行的状态 填了几个国王

方程很好想

但是需要预处理一下那些状态合法 以及两个状态之间是否可以相邻

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<vector>
#include<queue>
#define inf 2139062143
#define ll long long
#define MAXN 10
using namespace std;
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while(!isdigit(ch)) {if(ch=='-') f=-1;ch=getchar();}
    while(isdigit(ch)) {x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
int n,m,t,k[MAXN*60],cnt,num[MAXN*60];
ll dp[MAXN][MAXN*60][MAXN*MAXN];
bool ok[MAXN*60][MAXN*60];
void dfs(int pos,int f,int fst)
{
    k[++cnt]=pos,num[cnt]=f;
    for(int i=fst+2;i<=n;i++) dfs(pos+(1<<(i-1)),f+1,i);
}
int main()
{
    n=read(),m=read();
    dfs(0,0,-1);
    for(int i=1;i<=cnt;i++)
        for(int j=i;j<=cnt;j++)
            ok[i][j]=ok[j][i]=((k[i]&k[j])||((k[i]<<1)&k[j])||((k[j]<<1)&k[i]))?0:1;
    for(int i=1;i<=cnt;i++) dp[1][i][num[i]]=1LL;
    for(int g=2;g<=n;g++)
        for(int i=1;i<=cnt;i++)
            for(int d=num[i];d<=m;d++)
                for(int j=1;j<=cnt;j++)
                    if(ok[i][j]) dp[g][i][d]+=dp[g-1][j][d-num[i]];
    ll ans=0;
    for(int i=1;i<=cnt;i++) ans+=dp[n][i][m];
    printf("%lld",ans);
}