BZOJ1087 [SCOI2005]互不侵犯King

标签:状压DP

Description

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

Input

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

Output

　　方案数。

Sample Input

3 2

Sample Output

16

 

分析:设f[i][j][x]表示前i行放置了j的国王，且当前行状态（用二进制表示）为x时的方案数

判断状态是否符合要求可以枚举上一行状态，进行判断：如果当前行状态x和上一行状态y，符合下列任意条件，那么就是不符合要求的

X&(y<<1)

(x<<1)&y

X&y

预处理Num[x]，为状态为x时，该行放置国王的数量

 

Code

#include<bits/stdc++.h>
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define dep(i,a,b) for(int i=a;i>=b;i--)
#define mem(x,num) memset(x,num,sizeof x)
#define LL long long
using namespace std;
inline LL read()
{
    LL f=1,x=0;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
const int maxm=1006;
LL n,k,ans=0,f[16][maxm][maxm],num[maxm];
bool flag[maxm];
int main()
{
    n=read(),k=read();
    rep(i,0,(1<<n)-1)
        if(!(i&(i<<1))){
            flag[i]=true;
            int t=i;
            while(t){
                num[i]+=(t&1);
                t>>=1;
            }
            f[1][num[i]][i]=1;
        }
    rep(i,2,n)
        rep(j,0,k)
            rep(x,0,(1<<n)-1){
                if(!flag[x])continue;
                if(num[x]>j)continue;
                rep(y,0,(1<<n)-1){
                    if(!flag[y])continue;
                    if(x&y||x&(y<<1)||(x<<1)&y)continue;
                    f[i][j][x]+=f[i-1][j-num[x]][y];
                }
            }
    rep(i,0,(1<<n)-1)ans+=f[n][k][i];
    cout<<ans<<endl;
    return 0;
}