【计数DP】牛客小白月赛19

登录—专业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;
}