洛谷题单指南-状态压缩动态规划-P1879 [USACO06NOV] Corn Fields G
原题链接:https://www.luogu.com.cn/problem/P1879
题意解读:在m*n的区域种玉米,能种的是1不能种的是0,要求玉米不能种在相邻格子,求一共有多少种法,模100000000。
解题思路:
用状态压缩来表示一行玉米的种植情况,二进制中1表示种了,0表示没种。
对于一行状态,可以判断其合法性:相邻不能有连续1,在不能种的格子不能有1。
状态转移也很简单,通过枚举两行的合法状态,然后看两行状态是否存在上下相邻的1,不存在则可以转移。
状态表示:设f[i][j]表示前i行已种完,且第i行的状态为j的方案数
初始化:f[0][0] = 1,不种也是一种方案
状态转移:枚举所有第i行的状态j,枚举所有第i-1行的状态k,f[i][j] += f[i - 1][k]
结果:多枚举一行,枚举到第m+1行,答案是f[m + 1][0]
100分代码:
#include <bits/stdc++.h>
using namespace std;
const int N = 15, MOD = 1e8;
int mask[N];
int f[N][1 << 12];
vector<int> stat; //所有合法的状态
vector<int> mp[1 << 12]; //所有合法的状态转移
int m, n;
int main()
{
cin >> m >> n;
for(int i = 1; i <= m; i++)
{
for(int j = 0; j < n; j++)
{
int x;
cin >> x;
if(x == 0) mask[i] |= (1 << j);
}
}
//找到所有合法状态
for(int i = 0; i < 1 << n; i++)
{
if(i & i << 1) continue; //存在左右相邻的1
stat.push_back(i);
}
//找到所有合法的状态转移
for(auto i : stat)
{
for(auto j : stat)
{
if(i & j) continue; //存在上下相邻的1
mp[i].push_back(j);
}
}
f[0][0] = 1;
for(int i = 1; i <= m + 1; i++)
{
for(auto j : stat)
{
if(mask[i] & j) continue; //第i行存在不能种的土地
for(auto k : mp[j])
{
if(mask[i - 1] & k) continue; //第i - 1行存在不能种的土地
f[i][j] = (f[i][j] + f[i - 1][k]) % MOD;
}
}
}
cout << f[m + 1][0];
return 0;
}