7_22_C题 Batting Practice（求期望）

7_22_C题 Batting Practice

---

题意

连续投进若干球或连续不投进若干球则游戏结束，问游戏结束时的投球数的期望。

思路

设投进n球，投丢m球游戏结束，投不进的概率为p，进概率为q = 1-p，已连续投进x个球时剩余投球数的期望为in[x]，已连续不进x个时剩余投球数期望为out[x],则有
$in[x] = q * (in[x+1] + 1) + p * (out[1] + 1)$
同理
$out[x] = p * (out[x+1] + 1) + q * (in[1] + 1)$

展开化简得
$in[0] = out[0] = q * (in[1] + 1) * p (out[1] + 1)$
$in[1] = (q * out[1] + 1) * (q^{m-1} + q^{m-2} +…+1)$
$out[1] = (p * in[1] + 1) * (p^{n-1} + p^{n-2} +…+1)$
最后随便带入化简就行了。

代码

#include <bits/stdc++.h>
using namespace std;
const double esp = 1e-10;

int main(){
    int T,kas =1;
    scanf("%d",&T);
    while(T--){
        double p,q;
        int k1,k2;
        scanf("%lf %d %d",&p, &k1, &k2);
        q = 1-p;
        if(p < esp) printf("Case %d: %.9f\n",kas++,(double)k1);
        else if(q < esp) printf("Case %d: %.9f\n",kas++,(double)k2);
        else{
            double sum1 = 1 - pow(q, k1 - 1);
            double sum2 = 1 - pow(p, k2 - 1);
            double tmp1 = sum1 / p;
            double tmp2 = sum2 / q;
            double in = (tmp1 * sum2 +tmp2) / (1 - sum1*sum2);
            double out = tmp1 * (p * in + 1);
            printf("Case %d: %.9f\n",kas++,p*in+q*out + 1);
        }
    }
}