BZOJ1087【状压DP】

题目链接【http://www.lydsy.com/JudgeOnline/problem.php?id=1087】

题意：在N×N的棋盘里面放K个国王，使他们互不攻击，共有多少种摆放方案。国王能攻击到它上下左右，以及左上左下右上右下八个方向上附近的各一个格子，共8个格子。输入包含两个数N，K （ 1 <=N <=9, 0 <= K <= N * N），输出方案数。

解法：状压DP,首先处理每一行的有效状态，虽然有2^9种状态，但是同一行的有效状态不超过90，我们定义dp[i][t][x]表示：第i行的状态是x,并且一共用了t个国王的方案数。

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
LL dp[][][];
LL sta[], num[], cnt;
int n, k;
void stacnt()
{
    int mask = ( << n) - ;
    for(int i = ; i <= mask; i++)
    {
        if((i & (i << )) || (i & (i >> ))) continue;
        sta[++cnt] = i;
        int t = i;
        num[cnt] = __builtin_popcount(i);
    }
}
bool attrack(int x, int y)
{
    if(x & y) return false;
    if(x & (y << )) return false;
    if(x & (y >> )) return false;
    return true;
}
int main ()
{
    scanf("%d%d", &n, &k);
    stacnt();
    for(int i = ; i <= cnt; i++)//预处理第一行的状态
        dp[][num[i]][i] = ;
    for(int i = ; i <= n; i++)//第i层
        for(int t = ; t <= k; t++)//一共用了多少个士兵
            for(int x = ; x <= cnt; x++)//第i层的状态
                for(int y = ; y <= cnt; y++)//第i-1层的状态
                {
                    if(t - num[x] - num[y] >=  && attrack(sta[x], sta[y]))
                        dp[i][t][x] += dp[i - ][t - num[x]][y];
                }
    LL ans = ;
    for(int i = ; i <= cnt; i++)
        ans += dp[n][k][i];
    printf("%lld\n", ans);
    return ;
}