bzoj 1087 [SCOI2005]互不侵犯King 状态压缩dp

1087: [SCOI2005]互不侵犯King

Time Limit: 10 Sec  Memory Limit: 162 MB
[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

思路：状态压缩入门题；

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define esp 1e-13
const int N=5e2+,M=1e6+,inf=1e9+,mod=;
ll dp[][N][];
int check(int t)
{
    int num=t&(t<<);
    if(num==)
    return ;
    return ;
}
int check2(int i,int t)
{
    if((i&t)==&&(i&(t>>))==&&(i&(t<<))==)
    return ;
    return ;
}
int get1(int t)
{
    int sum=;
    while(t)
    {
        if(t&)
        sum++;
        t>>=;
    }
    return sum;
}
int main()
{
    int x,y,i,z,t;
    scanf("%d%d",&x,&y);
    dp[][][]=;
    int u=;
    for(i=;i<x;i++)
    u*=;
    for(i=;i<=x;i++)
    {
        for(t=;t<u;t++)
        {
            if(!check(t))
            continue;
            for(int j=;j<u;j++)
            {
                if(!check(j))
                continue;
                if(!check2(j,t))
                    continue;
                int hh=get1(t);
                for(int k=;k+hh<=y;k++)
                dp[i][t][k+hh]+=dp[i-][j][k];
            }
        }
    }
    ll ans=;
    for(i=;i<u;i++)
    ans+=dp[x][i][y];
    printf("%lld\n",ans);
    return ;
}