BZOJ 1087 互不侵犯King （状压dp）

1087: [SCOI2005]互不侵犯King

Time Limit: 10 Sec Memory Limit: 162 MB

Description

　　在N×N的棋盘里面放K个国王，使他们互不攻击，共有多少种摆放方案。国王能攻击到它上下左右，以及左上
左下右上右下八个方向上附近的各一个格子，共8个格子。

Input

　　只有一行，包含两个数N，K （ 1 <=N <=9, 0 <= K <= N * N）

Output

方案数。

Sample Input

3 2

Sample Output

16

思路：
对于每一行的状态，我们用1表示国王，0表示不放国王。
先枚举每行所有可能的状态，记录下合法状态。然后判断两个状态之间能否转移。若是可以，就更新dp。最后将最后一行的所有状态的方案数相加得ans。

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define LL long long
#define N 10
#define M 1200
using namespace std;

int n, m, lim;
LL f[N][M][N*N];//f[i][j][k]表示第i行状态为j时，有k个King时的方案数。
int str[M], num[M], cnt;

bool check(int S, int T){
    return ((S&T) | (S&(T<<1)) | (S&(T>>1)));
    //对于S中每个1（King）其下方三个位置只要有1（King）则不合法 
}

int main(){
    f[0][1][0] = 1;
    scanf("%d%d", &n, &m);
    lim = (1<<n)-1;//限度 
    for(int i=0; i<=lim; i++){
        if(( i&(i<<1) ) == 0){//没有两个国王相邻 
            str[++cnt] = i;//储存状态 
            int rstate = i, cc = 0;
            while( rstate ){
                cc += (rstate & 1);
                rstate >>= 1;
            }
            num[cnt] = cc;//当前状态行上的King的个数 
        }
    }
    for(int i=1; i<=n; i++){
        for(int j=1; j<=cnt; j++){
            for(int k=1; k<=cnt; k++)
            if( !check(str[j], str[k]) ){//判断j，k状态间是否能够转移 
                for(int l=num[k]; l<=m; l++){//不能少于当前行个数 
                    f[i][k][l] += f[i-1][j][l-num[k]];
                }
            }
        }
    }
    LL ans = 0;
    for(int i=1; i<=cnt; i++)
        ans += f[n][i][m];
    cout<<ans<<endl;
    return 0;
}