蓝桥杯两道我愿称为全是计数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;
}