CodeForces 296 B.Yaroslav and Two Strings（dp）

Description

给出两个长度均为$n$的数字串$a,b$ ，其中一些位置是$'?'$表示该位置数字不确定，要求确定这些位置的数字使得位置$i,j$，使得$a[i]>a[i],a[j]<b[j]$，问方案数

Input

第一行一个整数$n$表示两个串的串长，之后输入两个数字串$(1\le n\le 10^5)$

Output

输出方案数，结果模$10^9+7$

Sample Input

2
90
09

Sample Output

1

Solution

$dp[i][j][k]$表示从第一个位置到第$i$个位置，第一个串比第二个串大的位置是否出现$(j=1$表示出现过，$j=0$表示没出现过$)$，第二个串比第一个串大的位置是否出现$(k=1$表示出现过，$k=0$表示没出现过$)$，对于两个数字都确定的位置直接比较大小转移，对于一个确定一个不确定的位置根据不确定位置与确定位置数字的大小关系也很好转移，对于两个都不确定的位置，有$C_{10}^2$种情况使得该位置第一个串大于第二个串，有$C_{10}^2$种情况使得该位置第一个位置小于第二个串，有$10$种情况使得该位置两个串相同，以此转移，答案即为$dp[n][1][1]$

Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
const int INF=0x3f3f3f3f,maxn=100005;
#define mod 1000000007
int dp[maxn][2][2],n;
char a[maxn],b[maxn]; 
void add(int &x,int y)
{
    x=x+y>=mod?x+y-mod:x+y; 
}
int main()
{
    scanf("%d",&n);
    scanf("%s%s",a+1,b+1);
    for(int i=1;i<=n;i++)
    {
        if(a[i]!='?')a[i]-='0';
        if(b[i]!='?')b[i]-='0';
    }
    dp[0][0][0]=1;
    for(int i=1;i<=n;i++)
        for(int j=0;j<=1;j++)
            for(int k=0;k<=1;k++)
            {
                if(a[i]=='?'&&b[i]=='?')
                {
                    add(dp[i][j|1][k],(ll)45*dp[i-1][j][k]%mod);
                    add(dp[i][j][k|1],(ll)45*dp[i-1][j][k]%mod);
                    add(dp[i][j][k],(ll)10*dp[i-1][j][k]%mod);
                }
                else if(a[i]=='?')
                {
                    add(dp[i][j|1][k],(ll)(9-b[i])*dp[i-1][j][k]%mod);
                    add(dp[i][j][k|1],(ll)b[i]*dp[i-1][j][k]%mod);
                    add(dp[i][j][k],dp[i-1][j][k]);
                }
                else if(b[i]=='?')
                {
                    add(dp[i][j|1][k],(ll)a[i]*dp[i-1][j][k]%mod);
                    add(dp[i][j][k|1],(ll)(9-a[i])*dp[i-1][j][k]%mod);
                    add(dp[i][j][k],dp[i-1][j][k]);
                }
                else
                {
                    if(a[i]>b[i])add(dp[i][j|1][k],dp[i-1][j][k]);
                    else if(a[i]<b[i])add(dp[i][j][k|1],dp[i-1][j][k]);
                    else add(dp[i][j][k],dp[i-1][j][k]);
                }
            }
    printf("%d\n",dp[n][1][1]);
    return 0;
}