题意:
解法:
对于每个位置,如果s[i]!=0&&t[i]!=0&&s[i]!=t[i],
那么可以计算出当前位置s[i]>t[i]的概率,乘上s[1,i-1]与t[1,i-1]相等的概率P,则ans+=P.
从左往右递推,维护左边相等的概率即可.
code:
#include<bits/stdc++.h>
#define int long long
using namespace std;
const int maxm=2e6+5;
const int mod=1e9+7;
int s[maxm];
int t[maxm];
int n,m;
int ppow(int a,int b,int mod){
int ans=1%mod;a%=mod;
while(b){
if(b&1)ans=ans*a%mod;
a=a*a%mod;
b>>=1;
}
return ans;
}
void solve(){
cin>>n>>m;
for(int i=1;i<=n;i++)cin>>s[i];
for(int i=1;i<=n;i++)cin>>t[i];
int ans=0;
int eq=1;
int inv=ppow(m,mod-2,mod);
for(int i=1;i<=n;i++){
if(s[i]&&t[i]){
if(s[i]<t[i])break;
else if(s[i]>t[i]){
ans=(ans+eq)%mod;
break;
}
}else if(s[i]){
ans=(ans+(s[i]-1)*inv%mod*eq%mod)%mod;
eq=eq*inv%mod;
}else if(t[i]){
ans=(ans+(m-t[i])*inv%mod*eq%mod)%mod;
eq=eq*inv%mod;
}else{//!s[i]&&!t[i]
int cnt=(m-1)*m/2%mod;
ans=(ans+cnt*inv%mod*inv%mod*eq%mod)%mod;
eq=eq*inv%mod;
}
}
ans=(ans%mod+mod)%mod;
cout<<ans<<endl;
}
signed main(){
ios::sync_with_stdio(0);cin.tie(0);
solve();
return 0;
}