【XSY1528】azelso 概率&期望DP

题目大意

  有一条很长很长的路(出题人的套路),你在(0)位置,你要去(h)位置。

​  路上有一些不同的位置上有敌人,你要和他战斗,你有(p)的概率赢。若你赢,则你可以走过去,否则你会死。还有很多个重生点。你每经过一个重生点有(p)的概率插旗。你死亡后你会在最后一个插旗的位置重生,然后该位置的旗子消失。如果没有旗子,则你在(0)位置重生。

​  求你走到目的地的期望路程。模({10}^9+7)

​  (nleq 100000)

题解

​  这道题我用的式子和题解的不一样,最后推出来同一个式子。

​  设(E_i=)(i)段的期望通过次数,(f_i=)结算完第(i)个事件后不回到第(i)个点直接到达终点的概率。

​  根据期望与概率的关系,有:(E_i=frac1{f_i})

​  设(p=frac{a_{i+1}}{b_{i+1}})(即下一个事件发生的概率)

​  若下一个事件是'(X)'(敌人):

[f_i=pf_{i+1} ]

[frac1{E_i}=frac1{pE_{i+1}} ]

[E_i=frac{E_{i+1}}p ]

​  若下一个事件是'(F)'(重生点):

[f_i=f_{i+1}(1+p(1-f_{i+1})+p^2{(1-f_{i+1})}^2+cdots)=f_{i+1}frac1{1-p(1-f_{i+1})}=f_{i+1}frac1{1-p+pf_{i+1}} ]

​  (第一次成功+第一次插旗&失败&第二次成功+前两次插旗&失败&第三次成功...)

[frac1{E_i}=frac1{E_{i+1}}frac1{1-p+pfrac1{E_{i+1}}} ]

[frac1{E_i}=frac1{E_{i+1}-pE_{i+1}+p} ]

[E_i=(1-p)E_{i+1}+p ]

  然后把期望通过次数( imes)这段的长度加起来就好了。

​  时间复杂度:(O(n))

代码

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cstdlib>
#include<ctime>
#include<utility>
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;
ll p=1000000007;
ll fp(ll a,ll b)
{
	ll s=1;
	while(b)
	{
		if(b&1)
			s=s*a%p;
		a=a*a%p;
		b>>=1;
	}
	return s;
}
ll c[100010],a[100010],b[100010],d[100010];
ll f[100010];
int main()
{
	ll h,n;
	scanf("%lld%lld",&h,&n);
	int i;
	char s[10];
	for(i=1;i<=n;i++)
	{
		scanf("%s",s);
		if(s[0]=='X')
			c[i]=1;
		else
			c[i]=2;
		scanf("%lld%lld%lld",&d[i],&a[i],&b[i]);
		a[i]=a[i]*fp(b[i],p-2)%p;
	}
	f[n]=1;
	for(i=n;i>=1;i--)
		if(c[i]==1)
			f[i-1]=f[i]*fp(a[i],p-2)%p;
		else
			f[i-1]=((1-a[i]+p)%p*f[i]%p+a[i])%p;
	d[0]=0;
	d[n+1]=h;
	ll ans=0;
	for(i=0;i<=n;i++)
		ans=(ans+(d[i+1]-d[i])%p*f[i]%p+p)%p;
	printf("%lld
",ans);
	return 0;
}
原文地址:https://www.cnblogs.com/ywwyww/p/8511141.html