国王【状压dp】

最新推荐文章于 2022-11-24 17:07:42 发布

原创最新推荐文章于 2022-11-24 17:07:42 发布 · 626 阅读

1 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

动态规划

该博客探讨了在一个n×n的棋盘上放置k个国王，使得他们无法互相攻击的方案数量。通过使用状态压缩动态规划（状压DP）方法，博主解释了如何设置状态f[i][j][k]来表示第i行、第j个状态和k个国王的方案数，并利用位运算判断不同状态的合法性。文章提供了样例输入和输出，以及数据范围，并给出了关键的解决方案思路。

题目描述

原题来自：SGU 223
在 $n \times n$ 的棋盘上放 $k$ 个国王，国王可攻击相邻的 $8$ 个格子，求使它们无法互相攻击的方案总数。

输入

只有一行，包含两个整数 $n$ 和 $k$ 。

输出

每组数据一行为方案总数，若不能够放置则输出 00。

样例输入

【样例输入1】
3 2
【样例输入2】
4 4

样例输出

【样例输出1】
16
【样例输出2】
79

数据范围与提示

对于全部数据， $1≤n≤10,0≤k≤n^2$ 。

题解：

技巧：一般状压的题数据都很小
　设置状态 $f [i] [j] [k]$ 表示第i行是第j个状态有k个国王时的方案数。那如何判断上一行和这一行的状态合法？我们就可以用位运算，题目上说了上一行正上方不能放国王。我们将这一行的状态i,和上一行的状态i-1进行与运算，只要为真那么我们就不能选取，而且上一行右边和左边如果放国王的话，这一行同样不能放。所以我们可以先将上一位的状态左移（右移）后再进行与运算，只要结果为真就跳过这一行这个状态。

$C o d e :$

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int N=1005;
int n,m,cnt;
int a[N],sum[N];
ll f[11][N][105];
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=0;i<(1<<n);i++)
	{
        if(i&(i<<1)||i&(i>>1)) continue;
        a[++cnt]=i;
        int x=i;
        while(x) sum[cnt]+=x&1,x>>=1;  //记录每一行二进制下有多少个1
    }
    for(int i=1;i<=cnt;i++) f[1][i][sum[i]]=1;
    for(int i=2;i<=n;i++)
        for(int j=1;j<=cnt;j++)
            for(int k=1;k<=cnt;k++)
			{
                if(a[j]&a[k]||(a[j]>>1)&a[k]||(a[j]<<1)&a[k]) continue; 
                for(int l=sum[j]+sum[k];l<=m;l++)
                    f[i][j][l]+=f[i-1][k][l-sum[j]];
            }
    ll ans=0;
    for(int i=1;i<=cnt;i++)
        ans+=f[n][i][m];
    cout<<ans<<endl;
}