登录—专业IT笔试面试备考平台_牛客网
题意
思路
首先做法一定是计数 dp
然后状态设计,先设 dp[i]
然后看影响决策的因素:两边的火焰情况,那就 dp[i][0/1][0/1]表示 前 i 个,该位有无火焰,该位右边有无火焰的方案数
在状态设计的时候一定要体现该位的状态
这样设状态也可以算贡献
然后一定就是分类讨论了
#include <bits/stdc++.h>
constexpr int N = 1e6 + 10;
constexpr int mod = 1e9 + 7;
int dp[N][5][5];
/*前 i 个位置, 第 i 个位置是否有 *, 第 i + 1个位置是否有 * 的方案数*/
void solve() {
std::string s;
std::cin >> s;
int n = s.size();
s = " " + s;
dp[0][0][0] = dp[0][0][1] = 1;
for (int i = 1; i <= n; i ++) {
if (s[i] == '*') {
//-*
dp[i][1][0] += dp[i - 1][0][1] + dp[i - 1][1][1];
dp[i][1][0] %= mod;
//**
dp[i][1][1] += dp[i - 1][0][1] + dp[i - 1][1][1];
dp[i][1][1] %= mod;
}else if (s[i] == '0') {
//---
dp[i][0][0] += dp[i - 1][0][0];
dp[i][0][0] %= mod;
}else if (s[i] == '1') {
//--*
dp[i][0][1] += dp[i - 1][0][0];
dp[i][0][1] %= mod;
//*--
dp[i][0][0] += dp[i - 1][1][0];
dp[i][0][0] %= mod;
}else if (s[i] == '2') {
//*-*
dp[i][0][1] += dp[i - 1][1][0];
dp[i][0][1] %= mod;
}else if (s[i] == '?') {
for (int j = 0; j < 2; j ++) {
for (int k = 0; k < 2; k ++) {
//?jk
dp[i][j][k] += (dp[i - 1][0][j] + dp[i - 1][1][j]) % mod;
dp[i][j][k] %= mod;
}
}
}
}
std::cout << (dp[n][1][0] + dp[n][0][0]) % mod << "\n";
}
signed main() {
std::ios::sync_with_stdio(false);
std::cin.tie(nullptr);
int t = 1;
while(t --) {
solve();
}
return 0;
}