ZOJ-3415 Zhou Yu

JXNU_SONG

于 2021-08-10 17:20:42 发布

阅读量199

点赞数

分类专栏：概率期望

本文链接：https://blog.csdn.net/qq_45863710/article/details/119573441

版权

博弈论期望值动态规划等比数列数学建模

关键词由CSDN通过智能技术生成

概率期望专栏收录该内容

14 篇文章 0 订阅

订阅专栏

Zhou Yu

题目链接

题目大意：我们有 $n$ 滴血， $m$ 个花色，每轮操作猜一个花色，猜中花色回一滴血，但血量上限为 $n$ ，猜错则扣血，扣到 $0$ 则游戏结束。现在求游戏结束的期望轮数。

如果我们一开始定义 $E_i$ 为当前血量为 $i$ ，游戏结束的轮数的话，我们会发现十分难转移状态，式子难以化简，所以我们可以换一个想法，我们设 $E_i$ 血量为 $i$ 变为血量为 $i - 1$ 的期望轮数。

那么首先我们有

$E_n=\frac{m-1}{m}+\frac{1}{m}(1+E_n)$

解得： $E_n=\frac{m}{m-1}$

若 $i \neq = n$ 则有： $E_i=\frac{1}{m}(f_i+f_{i+1})+1$

则有： $E_i=\frac{m}{(m-1)}+\frac{m}{(m-1)^2}+\frac{m}{(m-1)^3}+...+\frac{m}{(m-1)^{n-i+1}}$

我们设 $q=\frac{1}{m-1}$

实际上 $E_i$ 也就是一个等比数列的和，我们得其通项为： $a_i=m*\frac{q-q^{n+1}}{1-q}$
注意这里的我们用 $n - i$ 的项替换了 $i$ 的项，或者直接感性理解可以推出这个通项。

那么我们要求的答案实际上就是： $\sum\limits_{i=1}^na_i$

那么实际上求 $a_i$ 的前 $n$ 项和也就是再求一个等差和等比的通项。

最终我们就可以推出答案为：
$m×\frac{n×q-\frac{q^2×(1-q^n)}{1-q}}{1-q}$

注意特判 $m = 2$ 的情况：

$a n s = n \times (n + 1)$

#include <bits/stdc++.h>

using namespace std;

typedef long long ll;

const int N = 2e5 + 10;

int n, m;

double q;

double ksm(double base, int n) {
    double ans = 1;
    while(n) {
        if (n & 1) ans = ans * base;
        base = base * base;
        n >>= 1;
    }
    return ans;
}

int main() {
#ifndef ONLINE_JUDGE
    freopen("in.txt", "r", stdin);
    freopen("out.txt", "w", stdout);
#endif
    while(~scanf("%d%d", &n, &m)) {
        if (m == 2) {
            double ans = (1ll + n) * n;
            printf("%.8f\n", ans);
            continue;
        }
        q = 1.0 / (m - 1);
        double tmp1, tmp2, tmp3, ans;
        tmp1 = n * q;
        tmp2 = q * q * (1 - ksm(q, n)) / (1 - q);
        tmp3 = 1 - q;
        ans = (tmp1 - tmp2) / tmp3;
        printf("%.8f\n", m * ans);
    }
}