Codefoeces 404D Minesweeper 1D "小范围后效性"dp

题目链接：http://codeforces.com/problemset/problem/404/D

题意：类似于扫雷游戏，但只有行，有些格子是已知的，有些未知，‘？’表示未知，其余为已知。'0'表示左右都没有雷，'1'表示左右一共有一个雷，'2'表示左右一共有两个雷，‘*’表示本身就是雷。问合法的情况一共有多少种？（借的题意）

虽然看起来某个状态可能受后面的状态的影响，但后面能影响的状态不多，所以也把这种状态考虑到当前的所有状态内，消去"后效性"。

#include <iostream>   
#include <string>   
#include <cstring>   
#include <algorithm>   
#include <stdio.h>   
#include <cctype>   
#include <queue>   
#include <stdlib.h>   
#include <cstdlib>   
#include <math.h>   
#include <set>   
#include <vector>   
using namespace std; 
#define ll __int64
#define mod 1000000007

char s[3000005];
ll dp[1000005][5];
/*
0 这位置为0
1 这位置为1且左边是雷
2 这位置为1且右边是雷
3 这位置为2
4 这位置为雷

*/
int main(){
	ll i, j, k;
	while(~scanf("%s",s)){
		memset(dp[0], 0, sizeof(dp[0]));
		if(s[0]=='0')dp[0][0] = 1;
		else if(s[0]=='1')dp[0][2] = 1;
		else if(s[0]=='*')dp[0][4] = 1;
		else if(s[0]=='?')
			dp[0][0] = dp[0][2] = dp[0][4] = 1;

		for(i = 1; s[i]; i++){
			memset(dp[i], 0, sizeof(dp[i]));
			if(s[i] == '0'){
				dp[i][0] = dp[i-1][1] + dp[i-1][0];	dp[i][0] %= mod;
			}
			else if(s[i] == '1'){
				dp[i][1] = dp[i-1][4]; 
				dp[i][2] = dp[i-1][0] + dp[i-1][1];	dp[i][2] %= mod;
			}
			else if(s[i] == '2'){
				dp[i][3] = dp[i-1][4];
			}
			else if(s[i] == '*'){
				dp[i][4] = dp[i-1][2] + dp[i-1][3] + dp[i-1][4]; dp[i][4] %= mod;
			}
			else if(s[i] == '?'){
				dp[i][0] = dp[i-1][1] + dp[i-1][0];
				dp[i][1] = dp[i-1][4];
				dp[i][2] = dp[i-1][0] + dp[i-1][1];
				dp[i][3] = dp[i-1][4];
				dp[i][4] = dp[i-1][2] + dp[i-1][3] + dp[i-1][4];
				dp[i][0] %= mod; dp[i][4] %= mod; dp[i][2] %= mod;
			}
		}
		i--;
		ll ans = dp[i][0] + dp[i][1] + dp[i][4];
		printf("%I64d\n", ans%mod);
	}
	return 0;
}