bzoj1087 [SCOI2005]互不侵犯King

题目大意:询问一个n*n方阵中摆放k个互不相同的王的方案数
对于这种方阵计数的问题通常考虑设定压缩一个状态来表示每行或列的情况
根据这题再维护到第几行和摆放了多少王的信息$f_{i+1,j+s,k_2}+=f_{i,s,k_1}$

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fod(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
const int N=2<<10;
typedef long long ll;
bool ck1[N],ck2[N][N];
int cnt[N],n,K,s;
ll f[10][100][N],ans=0,all;
void pre()
{
    for(int i=0;i<(1<<n);i++)
        if((i&(i>>1))==0) {
            s=0;
            for(int j=0;j<n;j++)if((1<<j)&i)s++;
            cnt[i]=s;ck1[i]=1;
        }

    for(int i=0;i<(1<<n);i++)if(ck1[i])
        for(int j=0;j<(1<<n);j++)if(ck1[j])
            if((i&j)==0&&((i>>1)&(j))==0&&((j>>1)&i)==0)ck2[i][j]=1;
}
int main()
{
    scanf("%d%d",&n,&K);
    all=(1<<n)-1;
    pre();
    for(int i=0;i<(1<<n);i++)if(ck1[i])f[1][cnt[i]][i]=1;
    for(int i=1;i<n;i++)
       for(int j=0;j<(1<<n);j++)if(ck1[j])
          for(int k=0;k<(1<<n);k++)if(ck1[k]&&ck2[j][k])
                for(int s=cnt[j];s+cnt[k]<=K;s++)
                    f[i+1][s+cnt[k]][k]+=f[i][s][j];
    for(int i=0;i<(1<<n);i++) ans+=f[n][K][i];
    printf("%lld\n",ans);
    return 0;   
}