洛谷P3802 小魔女帕琪

小魔女帕琪

题目链接

题意：小魔女有 $7$ 种魔法，每种魔法可以释放 $a_i$ 次，现在小魔女会每次随机选择一个魔法施放，直到没有魔法可以释放。现在我们定义终极魔法为释放序列中，连续的 $7$ 个不同的魔法称为一次终极魔法，现问小魔女释放终极魔法的次数期望。

我们定义 $n$ 为一共要释放的魔法数量， $p[i]$ 为第 $i$ 次释放魔法的时候释放终极魔法的概率。那么我们现在尝试求某个 $p[i]$ 。那么实际上要在 $i$ 处释放出终极魔法，只要 $[i-6,i]$ 这一段区间都释放不同的魔法即可，因为是随机选择魔法，所以放置魔法的顺序没有影响，都是随机选择，让最终的序列变成一种样子。那么我们只要从 $n$ 个魔法里面挑出 $7$ 个不同的先放在 $[i-6,i]$ ，剩下的随便放即可。那么发生此种情况的概率为 $f[i]=7!\times \frac{a_1}{n}\times \frac{a_2}{n-1}\times ...\times \frac{a_7}{n-6}$ ，答案即是 $\sum _{i=6}^{n}f[i]$

#include <bits/stdc++.h>

using namespace std;

typedef long long ll;

ll fac_7 = 5040;

double a[10], n, ans = 5040;

int main() {
#ifndef ONLINE_JUDGE
    freopen("in.txt", "r", stdin);
    freopen("out.txt", "w", stdout);
#endif
    for (int i = 1; i <= 7; ++i) {
        cin >> a[i];
        n += a[i];
        if ((int)a[i] == 0) {
            printf("0.000");
            return 0;
        }
    }
    for (int i = 1; i <= 7; ++i) {
        ans *= a[i] / (n - i + 1);
    }
    ans *= (n - 6);
    printf("%.3f", ans);
}