【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

同学很早之前打表过了，orz

这题正解状压DP，f[i][s][k]表示第i行上一行的状态为S，选了k个

预处理出所有可行状态DP就好了

 #include<cstdio>
 #include<cstring>
 #define ll long long
 using namespace std;
 int gs[],a[][];
 bool pd[];
 int n,K;
 ll ans,f[][][];
 bool ok(int x,int y){
     for(int i=;i<=;i++) if(a[x][i]&&(a[y][i]||a[y][i-]||a[y][i+])) return ;
     return ;
 }

 int main(){
     scanf("%d%d",&n,&K);
     memset(pd,,sizeof(pd));
     for(int i=;i<=(<<n)-;i++){
         int tmp=i,cnt=;
         while(tmp){
             a[i][++cnt]=tmp&;gs[i]+=&tmp;
             tmp>>=;
         }
         for(int j=;j<=n;j++) if(a[i][j]==&&a[i][j+]==){
             pd[i]=;break;
         }
         if(pd[i]) f[][i][gs[i]]+=;
     }
     f[][][]=;
     for(int i=;i<=n;i++)
         for(int j=;j<=(<<n)-;j++){
             if(!pd[j]) continue;
             for(int k=;k<=(<<n)-;k++){
                 if(ok(j,k)){
                     for(int l=gs[k];l<=K;l++)
                     f[i][k][l]+=f[i-][j][l-gs[k]];
                 }
             }
         }
     for(int i=;i<=(<<n)-;i++)if (pd[i]) ans+=f[n][i][K];
     printf("%lld",ans);
 }