Little Kings

sgu223:http://acm.sgu.ru/problem.php?contest=0&problem=223

题意：n*n的格子放k个国王，一共有多少种放发。

题解：简单的状压DP。

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
int n,m;
const int N=(1<<11);
long long dp[2][330][11*11];
int num,s[N],ct[N];
void init(){
   memset(dp,0,sizeof(dp));
   memset(s,0,sizeof(s));
   memset(ct,0,sizeof(ct));
   num=0;
}
bool ok(int i){
    return (i&(i<<1))==0;
}
int counts(int k){
    int ans=0;
    while(k>0){
        ans+=k%2;
        k/=2;
    }
    return ans;
}
bool judge(int s1,int s2){
   if((s[s1]&s[s2])||((s[s1]<<1)&s[s2])||((s[s1]>>1)&s[s2]))return false;
   return true;
}
void solve(){
    for(int i=0;i<(1<<n);i++){
        if(ok(i)){
            num++;
            s[num]=i;
            ct[num]=counts(i);
        }
    }
}
int main(){
    while(~scanf("%d%d",&n,&m)){
        init();
        solve();
     for(int i=1;i<=num;i++){
            dp[1][i][ct[i]]=1;
        }
        for(int i=2;i<=n;i++){
            memset(dp[i&1],0,sizeof(dp[i&1]));
            for(int j=1;j<=num;j++){//这一行的状态
                for(int k=ct[j];k<=m;k++){//这一行的个数
                    for(int g=1;g<=num;g++){//上一行的状态
                        if(judge(j,g)){
                            dp[i&1][j][k]+=dp[(i-1)&1][g][k-ct[j]];
                        }
                    }
                }
            }
        }
    long long ans=0;
    for(int i=1;i<=num;i++){
        ans+=dp[n&1][i][m];
    }
      printf("%I64d\n",ans);
  }
}