题目链接
327. 玉米田 - AcWing题库高质量的算法题库https://www.acwing.com/problem/content/description/329/今天是状压dp的第二讲,玉米田。通过这两道题基本熟悉了状压dp的基本思路。
这道题与昨天小国王题目有些差异,小国王是八个方向上不能有第二个国王,本题只有四个方向,但本题通过输入状态,给出了每行特定的非法位置,最后再套用模板即可
第一步//预处理行内合法状态
for(int i=0;i<1<<m;i++)
if(!(i&i>>1))
state.push_back(i);//预处理行内合法状态
第二步//预处理行间合法状态
for(int i=0;i<state.size();i++){
for(int j=0;j<state.size();j++){
int a=state[i],b=state[j];
if(!(a&b))
head[i].push_back(j);//预处理行间合法状态
}
}
第三步//枚举每行合法状态,再进行转移
for(int i=1;i<=n+1;i++){//枚举每行
for(int j=0;j<state.size();j++){
if(!(state[j]&w[i])){//判断每行合法稻田
for(int k:head[j])
f[i][j]=(f[i][j]+f[i-1][k])%mod;
}
}
}
状压dp基本就是这三步,再根据题意更改一些细节即可
ac代码为
#include<bits/stdc++.h>
using namespace std;
const int N = 14, M = 1 << 12, mod = 1e8;
int n, m;
int w[N];//存储规定非法的非法状态
vector<int> state;
vector<int> head[M];
int f[N][M];
int main(){
cin>>n>>m;
for(int i=1;i<=n;i++){
for(int j=0;j<m;j++){
int t;cin>>t;
w[i]+=(!t)*(1<<j);
}
}
for(int i=0;i<1<<m;i++)
if(!(i&i>>1))
state.push_back(i);//预处理行内合法状态
for(int i=0;i<state.size();i++){
for(int j=0;j<state.size();j++){
int a=state[i],b=state[j];
if(!(a&b))
head[i].push_back(j);//预处理行间合法状态
}
}
f[0][0]=1;
for(int i=1;i<=n+1;i++){//枚举每行
for(int j=0;j<state.size();j++){
if(!(state[j]&w[i])){//判断每行合法稻田
for(int k:head[j])
f[i][j]=(f[i][j]+f[i-1][k])%mod;
}
}
}
cout<<f[n+1][0];
}