DP mine

问题 A: mine
时间限制: 1 Sec 内存限制: 256 MB
提交: 129 解决: 31
[提交][状态]
题目描述
有一个1 维的扫雷游戏，每个格子用*表示有雷，用0/1/2 表示无雷并且相邻格子中有0/1/2 个雷。
给定一个仅包含?、、0、1、2 的字符串s，问有多少种方法将所有的?改为/0/1/2 使其合法。
输入
一行一个字符串s
输出
一行一个整数表示答案，对10^9+7 取模
样例输入
?1?
样例输出
2
提示
对于30%的数据，|S|<=20。

对于60%的数据，|S|<=1000。

对于100%的数据，|S|<=10^6。

通过数据范围发现效率应该是O(N)的。
然后发现‘1’非常恶心。其实1可以分开考虑，即分成雷在前面和雷在后面两种。讨论起来就非常方便了。

#pragma GCC optimize("O3")#include<cstdio>#include<cstdlib>#include<cstring>#include<iostream>#include<algorithm>#define N 1000005#define ll long long#define mod 1000000007using namespace std;int n;ll f[N][5];char s[N];int main(){       scanf("%s",s+1);n=strlen(s+1);    if(s[1]=='0')f[1][0]=1;    if(s[1]=='1')f[1][4]=1;    if(s[1]=='*')f[1][3]=1;    if(s[1]=='?')    {        f[1][0]=1;f[1][4]=1;f[1][3]=1;    }    for(int i=2;i<=n;i++)    {        if(s[i]=='?')        {            f[i][3]=(f[i-1][4]+f[i-1][2]+f[i-1][3])%mod;            f[i][0]=(f[i-1][0]+f[i-1][1])%mod;            f[i][1]=f[i-1][3];            if(i!=n)f[i][4]=(f[i-1][1]+f[i-1][0])%mod;            if(i!=n)f[i][2]=f[i-1][3];        }        else        {            if(s[i]=='*')f[i][3]=(f[i-1][4]+f[i-1][2]+f[i-1][3])%mod;            if(s[i]=='0')f[i][0]=(f[i-1][0]+f[i-1][1])%mod;            if(s[i]=='1')            {                f[i][1]=f[i-1][3];                f[i][4]=(f[i-1][1]+f[i-1][0])%mod;            }            if(s[i]=='2')f[i][2]=f[i-1][3];        }    }    ll ans=0;    for(int i=0;i<=3;i++)ans+=f[n][i];    cout<<ans%mod;}