计数dp cf404d

Posted 2022-04-16 iuk11

蓝桥杯两道我愿称为全是计数dp，昨天找了一道d，不敢练f。
计数dp就是找规律，把情况找全进行转移即可。
题目
题意见题目。
一开始设的情况不够用，后面补充全了。

• dp[i][0] 到第i位结尾，当前位为0
• dp[i][1] 到第i位结尾，当前位为1，且s[i-1]是一个雷
• dp[i][2] 到第i位结尾，当前位为1，且s[i-1]不是一个雷
• dp[i][3] 到第i位结尾，当前位为2
• dp[i][4] 到第i位结尾，当前位为雷

如果是?就把?可以替代的情况都填一遍。

#include<bits/stdc++.h>
using namespace std;
const int N=1e6+100;
const int mod=1e9+7;
typedef long long ll;
ll dp[N][5];
int main()
    string s;
    cin>>s;
    ll n=s.size();
    s="#"+s;
    if(s[1]=='0')
        dp[1][0]=1;
    else if(s[1]=='1')
        dp[1][2]=1;
    else if(s[1]=='2')
        ;
    else if(s[1]=='*')
        dp[1][4]=1;
    else
        dp[1][0]=dp[1][2]=dp[1][4]=1;
    
    for(int i=2;i<=n;i++)
        if(s[i]=='0'||s[i]=='?')
            dp[i][0]=(dp[i][0]+dp[i-1][0])%mod;
            dp[i][0]=(dp[i][0]+dp[i-1][1])%mod;
        
        if(s[i]=='1'||s[i]=='?')
            dp[i][1]=(dp[i][1]+dp[i-1][4])%mod;
            dp[i][2]=(dp[i][2]+dp[i-1][0])%mod;
            dp[i][2]=(dp[i][2]+dp[i-1][1])%mod;
        
        if(s[i]=='2'||s[i]=='?')
            dp[i][3]=(dp[i][3]+dp[i-1][4])%mod;
        
        if(s[i]=='*'||s[i]=='?')
            dp[i][4]=(dp[i][4]+dp[i-1][2])%mod;
            dp[i][4]=(dp[i][4]+dp[i-1][3])%mod;
            dp[i][4]=(dp[i][4]+dp[i-1][4])%mod;
        
    
    ll ans=0;
    ans=(ans+dp[n][0])%mod;
    ans=(ans+dp[n][1])%mod;
    ans=(ans+dp[n][4])%mod;
    cout<<ans<<endl;
    return 0;