SGU 223 Little Kings

题目链接：http://acm.sgu.ru/problem.php?contest=0&problem=223

题意：一个N*N的棋盘，放置M个国王，使得M个国王不能相互攻击，国王的攻击范围为与自己相邻的八个位置。问有多少放置方法。

分析：状态压缩DP。与POJ 1185  炮兵阵地 相似。但与之不同的是这里没有地形的限制，且当前行的状态只与前一行的状态和当前行之前所放置的国王总数有关。因此可以将这两种限制因素作为DP的状态。状态转移方程：dp[i][j][s1]=sum(dp[i-1][nm][s2])。dp[i][j][s1]指1至i行所放置的国王总数为j，状态为s1时的放置方法总数。

Code：

#include <algorithm>
#include <iostream>
#include <cstring>
#include <string>
#include <cstdio>
#include <vector>
#include <queue>
#include <cmath>
#include <map>
#include <set>
#define LL long long
#define Max(a,b) ((a)>(b)?(a):(b))
#define Min(a,b) ((a)<(b)?(a):(b))
using namespace std;

int state[150],num[150];
LL dp[15][110][150];
int tot,n,m;

void init_state(){
    tot=0;
    for(int i=0;i<(1<<n);i++){
        if(i&(i<<1)) continue;
        int t=i;
        num[tot]=0;
        while(t){
            num[tot]+=(t&1);
            t>>=1;
        }
        state[tot++]=i;
    }
}

int main()
{
    scanf("%d %d",&n,&m);
    init_state();
    memset(dp,0,sizeof(dp));
    for(int s=0;s<tot;s++){
        if(num[s]>m) continue;
        dp[1][num[s]][s]++;
    }
    for(int i=2;i<=n;i++){
        for(int j=0;j<=m;j++){
            for(int s1=0;s1<tot;s1++){
                if(num[s1]>j) continue;
                int nm=j-num[s1];
                for(int s2=0;s2<tot;s2++){
                    if(num[s2]>nm) continue;
                    if(state[s1]&state[s2]) continue;
                    if(state[s1]&(state[s2]<<1)) continue;
                    if(state[s1]&(state[s2]>>1)) continue;
                    dp[i][j][s1]+=dp[i-1][nm][s2];
                }
            }
        }
    }
    LL ans=0;
    for(int s=0;s<tot;s++){
        if(num[s]>m) continue;
        ans+=dp[n][m][s];
    }
    cout<<ans<<endl;
    return 0;
}