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

关注

原创最新推荐文章于 2026-05-16 06:05:38 发布 337 阅读 AI 写同款· GEO 优化 ›

【BZOJ 1087】【SCOI 2005】互不侵犯King 【状压DP】题目跳转:http://www.lydsy.com/JudgeOnline/problem.php?id=1087Description　　在N×N的棋盘里面放K个国王，使他们互不攻击，共有多少种摆放方案。国王能攻击到它上下左右，以及左上左下右上右下八个方向上附近的各一个格子，共8个格子。Input　　只有一行，包含两个数N，K （ 1 <=N <=9, 0 <= K <= N * N）Outp 阅读详情

sb状压，输int错了一次，没救了。。。

#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define N (1<<9)+10
int n,m;
bool g[N][N];
ll f[11][110][N],ans;
int num[N];
int check(int x,int y)
{
    for(int i=0;i<n;i++)
        if(y>>i&1)
        {
            if(y>>i+1&1)return 0;
            if(x>>i+1&1)return 0;
            if(i&&x>>i-1&1)return 0;
            if(i&&y>>i-1&1)return 0;
            if(x>>i&1)return 0;
        }
    return 1;
}
int main()
{
    //freopen("tt.in","r",stdin);
    scanf("%d%d",&n,&m);
    for(int i=0;i<1<<n;i++)
        for(int j=0;j<1<<n;j++)
            g[i][j]=check(i,j);
    for(int i=1;i<1<<n;i++)
        num[i]=num[i-(i&-i)]+1;
    f[0][0][0]=1;
    for(int i=0;i<n;i++)
        for(int j=0;j<=m;j++)
            for(int k=0;k<1<<n;k++)
                for(int t=0;t<1<<n;t++)
                    if(g[k][t]&&j+num[t]<=m)
                        f[i+1][j+num[t]][t]+=f[i][j][k];
    for(int i=0;i<1<<n;i++)
        ans+=f[n][m][i];
    printf("%lld\n",ans);
    return 0;
}