用二进制表示状态,用十进制存储状态
状态压缩的对象必须只有两种状态:0/1,如果有三种状态用三进制表示也可以
即每个单元格只有两种状态,用0/1来表示,整个状态数据就是一串0和1组成的二进制数
但是用十进制数存储,即一行数据用一个十进制数就可以表示。所以开辟一个一维整形数组就可以存储一个二维的田地了
- 行内合法,枚举一行中的所有状态,用位运算筛选出合法状态,再用十进制存储到数组里
- dp的起始条件赋值
- 行间合法,一共好几层循环,最外面循环的枚举行(循环次数一般是n+以枚举第i行的合法状态开始多出来的枚举层数),往里依次是枚举其他状态、枚举第i行的合法状态、枚举第i-1行的合法状态、枚举第i-2行的合法状态……,用位运算判断状态转移的条件(通常是列,让列合法,即行间合法)
- 判断通过,则执行状态转移,每个类累加上一行的兼容类
- 输出的时候,输出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;
}
参考视频: