状态压缩动态规划讲解

最新推荐文章于 2025-04-26 13:43:52 发布

原创最新推荐文章于 2025-04-26 13:43:52 发布 · 365 阅读

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

【前言】

状态压缩动态规划，就是我们俗称的状压DP，是利用计算机二进制的性质来描述状态的一种DP方式，状压其实是一种很暴力的算法，因为他需要遍历每个状态，所以将会出现2^n的情况数量，不过这并不代表这种方法不适用：一些题目可以依照题意，排除不合法的方案，使一行的总方案数大大减少从而减少枚举

【预备知识】

在学习状压DP前，必须先要学习一下位运算
①~~左移：~~ 左移一位，就相当于某数乘以2，则左移n位，相当于某数乘以2ⁿ
②~~右移：~~ 右移一位，就相当于某数除以2，则左移n位，相当于某数除以2ⁿ
③~~与运算：~~ 按位进行与运算，两数同位均为1则对应位置为1，否则为0
④~~或运算：~~ 按位进行或运算，两数同位均为0则对应位置为0，否则为1
⑤~~非运算：~~ 按位取反

【状压题目】——国王

~~链接：http://ybt.ssoier.cn:8088/problem_show.php?pid=1592~~

【题目描述】
在 n×n 的棋盘上放 k 个国王，国王可攻击相邻的 8 个格子，使它们无法互相攻击的方案总数。

【输入】
只有一行，包含两个整数 n 和 k。

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

【输入样例】

3 2

【输出样例】

【分析】

根据数据可知这是一道状压DP，根据我们所学的状压知识解决即可

首先，我们要进行预处理，数每个二进制位上有多少个1，再逐渐判断国王状态是否合法

【代码】

#include<bits/stdc++.h>
using namespace std;
typedef long long ll; 
int n,m;
ll ans;
int k[(1<<10)+5],ka[101];
ll f[20][(1<<10)+5][100];//行，状态，多少王 
bool bfs(int x)//判断状态是否合法（即不能在一行相邻） 
{
	if(k[x]>m) return 0;
	memset(ka,0,sizeof(ka));
	int d=x,cnt=0;
	while(d)
	{
		cnt++;
		if(d&1)
			ka[cnt]=1;
		d>>=1;
	}
	for(int i=1;i<=n;i++)
		if(ka[i]==1&&(ka[i-1]==1||ka[i+1]==1))
			return 0;
	return 1;
}
void count_king()//数二进制位上国王的个数 
{
	for(int i=0;i<=(1<<n)-1;i++)
	{
		int d=i,v=0;
		while(d)
		{
			if(d&1)
				v++;
			d>>=1;
		}
		k[i]=v;
	}
}
bool pd(int a,int b)//判断状态是否合法（即上下不能相邻） 
{
	if(a&b) return 0;
	if(a&(b<<1)) return 0;
	if((a<<1)&b)return 0;
	return 1;
}
int main()
{	
	cin>>n>>m;
	count_king();
	for(int i=0;i<=(1<<n)-1;i++)
		if(bfs(i))
			f[1][i][k[i]]=1;
	for(int i=2;i<=n;i++)
	{
		for(int j=0;j<=(1<<n)-1;j++)
		{
			if(!bfs(j)) continue;
			for(int kk=0;kk<=(1<<n)-1;kk++)
			{
				if(!pd(j,kk)) continue;
				for(int s=k[j];s<=m;s++)
					if(s-k[j]>=k[kk])
						f[i][j][s]+=f[i-1][kk][s-k[j]];
			}
		}
	}
	for(int j=1;j<=n;j++)
		for(int i=1;i<=(1<<n)-1;i++)
			if(bfs(i))ans+=f[j][i][m];
	cout<<ans; 
}