csuoj2275-Left-Right-Win(概率期望)

csuoj2275-Left-Right-Win(概率期望)

题目大意

给出一张圆桌，其上有n个位置.给定向左和向右转移的概率$P_{left}$,$P_{right}$.初始的判定点在0.在每一点判定点将进行判定，其将有$P_{left}$的概率将判定点左移一个位置,有$P_{right}$的概率右移一个位置，也有$1-P_{left}-P_{right}$的概率判中奖，并终止判定.判中奖的位置将得到100的奖励，问位置k的奖励的期望是多少？

解题思路

设位置i判为中奖的概率为dp[i]则可以列出转移方程：
$$dp[i]=P_{left}\ast dp[i-1]+P_{right}\ast dp[(i+1)\%n]$$
设为等式一

而初始点的中奖概率为
$$dp[0]=P_{left}\ast dp[n-1]+P_{right}\ast dp[1]+(1-P_{lrft}-P_{right})$$
设为等式二

通过前面的式子我们可以发现，所有点的中奖概率都可以通过其他两个点的概率线性组合得到.将dp[0],dp[1]设为这个线性系的基底，则可以设下式：
$$dp[i]=c[i]\ast dp[0]+d[i]\ast dp[1]$$
设为等式三

进一步地，由等式一我们可以通过变换得到
$$dp[i+1]=(1/P_{right})\ast dp[i]-(P_{left}/P_{right})\ast dp[i-1]$$
将等式三带入，则可以得到递推式
$$\begin{gathered} c[i]=(1/P_{right})\ast c[i-1]-(P_{left}/P_{right})\ast c[i-2] \\ d[i]=(1/P_{right})\ast d[i-1]-(P_{left}/P_{right})\ast d[i-2] \end{gathered}$$
其中c[0]=1,c[1]=0,d[0]=0,d[1]=1,故也就可以推出所有的c与d

而显然可以有
$$\sum _{i=0}^{n-1}dp[i]=1$$
故有
$$(\sum _{i=0}^{n-1}c[i])\ast dp[0]+(\sum _{i=0}^{n-1}d[i])\ast dp[1]=1$$
设为等式四

而根据等式二，我们可以通过变换得到
$$(1-P_{left}\ast c[n-1])\ast dp[0]+(P_{left}\ast d[n-1]-P_{right})\ast dp[1]=1-P_{left}-P_{right}$$
设为等式五

联立等式四、五，则可以解出$dp[0],dp[1]$

而最终的答案也就是
$$dp[k]=c[k]\ast dp[0]+d[k]\ast dp[1]$$

AC代码

#include<bits/stdc++.h>
using namespace std;
double lpro,rpro;
double c[20],d[20];
double solve(int n,int k,double lpro,double rpro)
{
	c[0]=1,c[1]=0;
	d[0]=0,d[1]=1;
	double sumc=1.0,sumd=1.0;
	for(int i=2;i<n;i++)
	{
		c[i]=(1/rpro)*c[i-1]-(lpro/rpro)*c[i-2];
		d[i]=(1/rpro)*d[i-1]-(lpro/rpro)*d[i-2];
		sumc+=c[i],sumd+=d[i];
	}
	double a=1-lpro*c[n-1],b=-lpro*d[n-1]-rpro;
	double e=sumc,f=sumd;
	double s1=1-rpro-lpro;
	double s2=1;
	double dp0=(s1*f-s2*b)/(a*f-e*b);
	double dp1=(s1*e-s2*a)/(b*e-f*a);
	return c[k]*dp0+d[k]*dp1;
}
int main()
{
	int t;
	scanf("%d",&t);
	for(int kace =1; kace <=t;kace++)
	{
		int no,n,k;
		scanf("%d%d%d%lf%lf",&no,&n,&k,&lpro,&rpro);
		printf("%d %.2f\n",kace ,100*solve(n,k,lpro,rpro));
	}
}