牛客练习赛39 C：流星雨 (概率dp+除法逆元)

【题目】

流星雨

【题解】

概率dp。
设 为第i天下雨的真正概率，由于第i天的概率只与第i-1天有关，所以有

复杂度：加上求逆元的复杂度，O(nlogn)。

【代码】

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int mod=1e9+7;
const int maxn=100005;
ll w[maxn],p[maxn],T[maxn];
ll n,a,b,P;
ll quick_pow(ll a,ll b)
{
    ll r=1,tmp=a;
    while(b)
    {
        if(b&1) r=r*tmp%mod;
        tmp=tmp*tmp%mod;
        b>>=1;
    }
    return r%mod;
}
int main()
{
	scanf("%lld%lld%lld",&n,&a,&b);
	P=1ll*a*quick_pow(b,mod-2)%mod;
	for(int i=1;i<=n;i++) scanf("%lld",&w[i]);
	for(int i=1;i<=n;i++)
	{
		scanf("%lld%lld",&a,&b);
		p[i]=a*quick_pow(b,mod-2)%mod;
	}
	T[1]=p[1];
	for(int i=2;i<=n;i++)
	{
	    T[i]+=T[i-1]*((p[i]+P)%mod)%mod;
		T[i]+=(1-T[i-1]+mod)%mod*p[i]%mod;
		T[i]%=mod;
	}
	ll ans=0;
	for(int i=1;i<=n;i++)
	{
	    ans+=1ll*T[i]*w[i]%mod;
	    ans%=mod;
	}
	printf("%lld\n",ans);
	return 0;
}