SCOI2005 BZOJ1087 互不侵犯KING

1087: [SCOI2005]互不侵犯King

Time Limit: 10 Sec  Memory Limit: 162 MB
Submit: 4601  Solved: 2668
[Submit][Status][Discuss]

Description

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

Input

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

Output

　　方案数。

Sample Input

3 2

Sample Output

16

状压dp入门题。

记 f[i][j] 为 第ｉ行状态为 j 的方案数， 发现第 i 行只和第i-1行有关

枚举合法状态 + 简单背包即可

#include <cstdio>
#include <cstring>
#include <algorithm> 
#define ll long long
using namespace std; 
int T, n, m;
int mp[8][8];
int sum[8][1<<10];
ll f[10][1<<10][90], ans;
int getsum(int sta){
    int sum = 0;
    while(sta){
        if(sta & 1) sum++;
        sta >>= 1; 
    }
    return sum;
}
int main(){
    int i, j, k, t, z;
    ans = 0;
    scanf("%d%d", &n, &m);
    f[0][0][0] = 1;
    for(i = 1; i <= n; i++){
        for(j = 0; j < (1<<n); j++){
            if(j & (j>>1)) continue;
            for(k = 0; k < (1<<n); k++){
                if(!(k & (k>>1)) && !(j & k) && !(j & (k>>1)) && !(j & (k<<1))){
                	z = getsum(k);
                	for(t = m; t >= z; t--) f[i][k][t] += f[i-1][j][t-z];
                }
            }
        }
    }
    for(j = 0; j < (1<<n); j++) ans += f[n][j][m];
    printf("%lld\n", ans);
    return 0;
}