矩阵计数-状压DP

0矩阵计数 - 蓝桥云课 (lanqiao.cn)

二进制表示状态,用十进制存储状态

状态压缩的对象必须只有两种状态:0/1,如果有三种状态用三进制表示也可以
即每个单元格只有两种状态,用0/1来表示,整个状态数据就是一串0和1组成的二进制数
但是用十进制数存储,即一行数据用一个十进制数就可以表示。所以开辟一个一维整形数组就可以存储一个二维的田地了

  1. 行内合法,枚举一行中的所有状态,用位运算筛选出合法状态,再用十进制存储到数组里
  2. dp的起始条件赋值
  3. 行间合法,一共好几层循环,最外面循环的枚举行(循环次数一般是n+以枚举第i行的合法状态开始多出来的枚举层数),往里依次是枚举其他状态、枚举第i行的合法状态、枚举第i-1行的合法状态、枚举第i-2行的合法状态……,用位运算判断状态转移的条件(通常是列,让列合法,即行间合法)
  4. 判断通过,则执行状态转移,每个类累加上一行的兼容类
  5. 输出的时候,输出dp[n+以枚举第i行的合法状态开始多出来的枚举层数][0][0]……

dp数组的一般写法:

dp[i][j][k] 表示第i行合法状态为j,第i-1行合法状态为k时,符合条件的数量

dp[i][j] 表示第i行合法状态为j,符合条件的数量

dp[i][j][k] 表示到第i行为止(前i行)已经放了j个国王,第i行合法状态为k,符合条件的数量

题目:

代码:

#include <bits/stdc++.h>
#define int long long
#define endl '\n'
using namespace std;
//字符O用二进制0表示,字符X用二进制1表示
//dp[i][j][k]表示第i行合法状态为j,第i-1行合法状态为k时,符合条件的矩阵数量
//状态转移方程考虑连续三行的情况,第i行状态为a,i-1行状态为b,i-2行状态为c
//如果state[a]&state[b]&state[c]==0,即三行状态的二进制数同列没有连续的1,则说明这一列不是连续的3个X
int dp[10][1<<6][1<<6];
//存一行中的合法状态
int state[1<<6];//1<<6代表的是1*2^6次幂,state数组是一行的合法状态集
int n,m,cnt;//cnt是一行的合法状态个数
bool check(int shu){//判断一行中有没有连续的三个1(即连续的三个X)
	int num=0;
	while(shu){
		if((shu&1)==1) num++;
		else num=0;
		if(num>=3) return true;
		shu=shu>>1;
	}
	return false;
}
void solve(){
	cin>>n>>m;
	for(int i=0;i<(1<<m);i++){//枚举一行的所有状态
		if(check(i)) continue;//如果这一行存在三个连续的1
		state[cnt++]=i;//把合法状态用十进制保存
	}
	dp[0][0][0]=1;
//	for(int i=0;i<cnt;i++){
//		for(int j=0;j<cnt;j++){
//			dp[0][state[i]][state[j]]=1;
//		}
//	}
	for(int i=1;i<=n+2;i++){//枚举行
		for(int a=0;a<cnt;a++){//枚举第i行的合法状态
			for(int b=0;b<cnt;b++){//枚举第i-1行的合法状态
				for(int c=0;c<cnt;c++){//枚举第i-2行的合法状态
					if(!(state[a]&state[b]&state[c])){//不存在同列的三个连续1
						dp[i][state[a]][state[b]]+=dp[i-1][state[b]][state[c]];//状态转移
					}
				}
			}
		}
	}
//	cout<<dp[n][0][0];
	cout<<dp[n+2][0][0];//只在1-n行放X(1),n+2什么都不放
}
signed main(){
	ios::sync_with_stdio(0);
	cin.tie(0);
	cout.tie(0);
	solve();
	return 0;
}

参考视频:

E25 状态压缩DP 小国王_哔哩哔哩_bilibili

E26 状态压缩DP 玉米田_哔哩哔哩_bilibili

E27 状态压缩DP 炮兵部队_哔哩哔哩_bilibili

  • 11
    点赞
  • 11
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值