牛客周赛 Round 60 F 期望dp

题目链接

我们设置p[i][1]为读第i个的时候，下一个会读i+1个的概率，p[i][2]为读第i个的时候，下一个会读第i个的概率，p[i][3]为读第i个的时候，下一个会读第i-1个的概率。

p数组可以预处理出来，这个直接算就是，就不过多描述了

然后我们规定dp[i]为从第i-1个开始读，能读到第i个的期望次数

对于第i-1个，有p1的概率，下一个能读到i+1，p2的概率还是自己，p3是读i-1

然后你就可以得到公式

化简一下就是

然后以此遍历来算就可以了，最后都加起来就是答案

#include <bits/stdc++.h>
using namespace std;
const int N = 1e6+5;
#define int long long
const int INF = 4e18+5;
#define F(i,l,r) for(int i=l;i<=r;i++)
#define R(i,l,r) for(int i=r;i>=l;i--)
#define vv vector
#define fi first
#define se second
#define pii pair<int,int>
typedef long long ll;
//const int mod = 998244353; 
const int mod = 1e9+7; 
const int M = 1e7+5;
int a[N],b[N],c,d,vis[N],pri[N],ans1[N],ans2[N],dp[N],p[N][4]; 
int dx[4]={1,0,-1,0};
int dy[4]={0,1,0,-1};
int n,m,k,ans,x,y,n0,n2,m0,m2,op,op1,op2;
string s;
map<pii,int>mp;
map<int,int>mmap;
//char mp[1005][1005];
//vector<int>v1[N],v2[N];
int lowbit(int x) {
	return x&-x;
}
char xx[N];
bool temp;
int kuai(int x,int y){
	int sum=1;
	x%=mod;
	while(y){
		if(y%2){
			y--;
			sum*=x;
			sum%=mod;
		  }
	  x*=x;
	  y/=2;
	  x%=mod;
	}
	return sum;
}
struct node{
	int shu;
	int id;
}jie[N];
bool cmp(node x,node y){
	return x.shu<y.shu;
}
void solve()
{
	cin>>n;//1是到下一个，2是到自身，3是到前一个 
	F(i,1,n-1) cin>>a[i]; F(i,1,n-1) cin>>b[i];
	p[1][1]=a[1]*kuai(a[1]+b[1],mod-2)%mod;
	p[1][2]=b[1]*kuai(a[1]+b[1],mod-2)%mod; 
	p[1][3]=0;
	F(i,2,n-1){
		int inv=kuai(a[i]+b[i],2);
		p[i][1]=a[i]*a[i]%mod*kuai(inv,mod-2)%mod;
		p[i][2]=2*a[i]*b[i]%mod*kuai(inv,mod-2)%mod;
		p[i][3]=b[i]*b[i]%mod*kuai(inv,mod-2)%mod;
	}
	//cout<<p[1][1]<<" "<<p[1][2]<<endl;
	dp[1]=1;  ans=0;
	F(i,2,n) dp[i]=(p[i-1][1]+p[i-1][2]+p[i-1][3]+p[i-1][3]*dp[i-1]%mod)%mod*kuai(((1-p[i-1][2]-p[i-1][3])%mod+mod)%mod,mod-2)%mod;
	F(i,1,n) {
		ans+=dp[i];
		ans%=mod;
        //cout<<dp[i]<<" ";
	}
	cout<<ans<<endl;
}
signed main() 
{
	ios::sync_with_stdio(false);
    cin.tie(0), cout.tie(0); // cin.tie(nullptr);
    int T = 1;                                                                                                                                                                                                                                                                    
    //cin >> T;
    for (int i = 1; i <= T; i++)
    {
        solve();
    }
    return 0;
}
/*
6 4 3 1 2
1 1 4 5 1 4
*/