Codeforces 891E Lust

最新推荐文章于 2021-11-15 22:27:18 发布

ypxrain

最新推荐文章于 2021-11-15 22:27:18 发布

阅读量292

点赞数 1

分类专栏：数学

本文链接：https://blog.csdn.net/ypxrain/article/details/79997129

版权

数学专栏收录该内容

22 篇文章

订阅专栏

本文介绍了一种计算特定随机序列操作后结果期望值的方法，通过生成函数和组合数学技巧，有效地解决了大规模数据集上的计算问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题意

现在给你一有n个整数的序列a[]，有一个初始为0的值res，重复下面的过程k次：
“随机选择一个[1,n]之间的下标x，res加上所有满足i≠x的a[i]的乘积，然后将a[x]减去1”
问最后res的期望值，对10^9+7取模
n<=5000
k<=10^9

分析

这个其实就是要求

\prod A - \prod A'

$\prod A-\prod A'$ 其中A表示初始状态的乘积，A’表示结束状态乘积的期望。
那么现在问题就是我们怎么求A’
我们设我们对

a[i] a [ i ] $a[i]$ 操作了

b[i] b [ i ] $b[i]$ 次，那么我们有

E = k ! \prod ( a [ i ] - b [ i ] ) n k \prod ( b [ i ] ) !

$E=\frac{k!\prod (a[i]-b[i])}{n^k\prod (b[i])!}$
我们考虑第i个的生成函数位

f i = \sum j \geq 0 a i - j j !

$f_i=\sum_{j\geq 0}\frac{a_i-j}{j!}$
于是总的生成函数就为

f (x) = \prod \sum j \geq 0 a i - j j ! x j

$f(x)=\prod \sum_{j\geq 0}\frac{a_i-j}{j!}x^j$

= \prod \sum a i x j j ! - x * x j - 1 ( j - 1 ) !

$=\prod \sum \frac{a_ix^j}{j!}-\frac{x*x^{j-1}}{(j-1)!}$

= e n x \prod (a i - x)

$=e^{nx}\prod (a_i-x)$
后面的我们就可以n^2处理了

代码

#include <bits/stdc++.h>

const int N = 5000 + 10;
const int MOD = 998244353;

int read()
{
    int x = 0, f = 1;
    char ch = getchar();
    while (ch < '0' || ch > '9') {if (ch == '-') f = -1; ch = getchar();}
    while (ch >= '0' && ch <= '9') {x = x * 10 + ch - '0'; ch = getchar();}
    return x * f;
}

int f[N][N];
int a[N];

int ksm(int x,int y)
{
    int res = 1;
    while (y)
    {
        if (y & 1)
            res = 1ll * x * res % MOD;
        x = 1ll * x * x % MOD;
        y >>= 1;
    }
    return res;
}

int main()
{
    freopen("manastorm.in","r",stdin);
    freopen("manastorm.out","w",stdout);
    int n = read(), k = read();
    for (int i = 1; i <= n; i++)
        a[i] = read();
    f[0][0] = 1;
    for (int i = 1; i <= n; i++)
    {
        f[i][0] = 1ll * f[i - 1][0] * a[i] % MOD;
        for (int j = 1; j <= i; j++) 
            f[i][j] = (1ll * f[i - 1][j] * a[i] - f[i - 1][j - 1]) % MOD;
    }
    int ans = 0;
    for (int i = 0; i <= std::min(n, k); i++)
    {
        int res = 1;
        for (int j = k - i + 1; j <= k; j++) 
            res = 1ll * res * j % MOD;
        ans = (ans + 1ll * f[n][i] * ksm(n, k - i) % MOD * res) % MOD;
    }
    ans = 1ll * ans * ksm(ksm(n, k), MOD - 2) % MOD;
    int m = 1;
    for (int i = 1; i <= n; i++)
        m = 1ll * m * a[i] % MOD;
    m = (m - ans + MOD) % MOD;
    printf("%d\n",m);
}