【Atcoder】 [ABC231G] Balls in Boxes

最新推荐文章于 2024-08-12 19:41:22 发布

Farmer_D

最新推荐文章于 2024-08-12 19:41:22 发布

阅读量127

点赞数

分类专栏： Atcoder 文章标签：算法

本文链接：https://blog.csdn.net/djx2021/article/details/134552524

版权

Atcoder 专栏收录该内容

29 篇文章 0 订阅

订阅专栏

题目链接

AT方向
 Luogu方向

题目解法

首先求期望变成计数
然后考虑生成函数，因为是有顺序的，所以用 $e g f$
考虑列出式子： $ans=\frac{k!}{n^k}[x^k]\sum\limits_{i=1}^n\sum\limits_{j=0}^{\infty}(a_i+j)\frac{x^j}{j!}$
化简，因为 $a_i$ 是定值， $j$ 可以和后面的 $\frac{1}{j!}$ 相乘变成 $\frac{1}{(j-1)!}$ ，所以拆开 $a_i+j$ ，可得
$ans=\frac{k!}{n^k}[x^k]\sum\limits_{i=1}^n\sum\limits_{j=0}^{\infty}(a_i\frac{x^j}{j!}+\frac{x^j}{(j-1)!})\\ =\frac{k!}{n^k}[x^k]\sum\limits_{i=1}^n\sum\limits_{j=0}^{\infty}(a_i+x)e^x\\ =\frac{k!}{n^k}[x^k]e^{nx}\sum\limits_{i=1}^n\sum\limits_{j=0}^{\infty}(a_i+x)\\ =\frac{k!}{n^k}[x^k](\sum\limits_{i=0}^{\infty}\frac{n^i}{i!})(\sum\limits_{i=1}^n\sum\limits_{j=0}^{\infty}(a_i+x))$
后面的 $\sum\limits_{i=1}^n\sum\limits_{j=0}^{\infty}(a_i+x)$ 可以 $d p$ 暴力卷，所以可以写成 $ans=\frac{k!}{n^k}[x^k](\sum\limits_{i=0}^{\infty}\frac{n^i}{i!})F(x)$
因为后面的 $F (x)$ 只有 $n$ 项，不妨直接枚举其次数
$ans=\frac{1}{n^k}\sum\limits_{i=0}^nF(x,i)n^{k-i}k^{\underline i}$
可以直接计算
时间复杂度 $O(n^2)$ ，使用分治 $n tt$ 代替 $d p$ 可以做到 $O(nlog^2n)$

#include <bits/stdc++.h>
#define F(i,x,y) for(int i=(x);i<=(y);i++)
#define DF(i,x,y) for(int i=(x);i>=(y);i--)
#define ms(x,y) memset(x,y,sizeof(x))
#define SZ(x) (int)x.size()-1
using namespace std;
typedef long long LL;
typedef unsigned long long ull;
typedef pair<int,int> pii;
template<typename T> void chkmax(T &x,T y){ x=max(x,y);}
template<typename T> void chkmin(T &x,T y){ x=min(x,y);}
const int N=1100,P=998244353;
int n,k,a[N];
inline int read(){
    int FF=0,RR=1;
    char ch=getchar();
    for(;!isdigit(ch);ch=getchar()) if(ch=='-') RR=-1;
    for(;isdigit(ch);ch=getchar()) FF=(FF<<1)+(FF<<3)+ch-48;
    return FF*RR;
}
int coef[N];
int qmi(int a,int b){
    int res=1;
    for(;b;b>>=1){
        if(b&1) res=1ll*res*a%P;
        a=1ll*a*a%P;
    }
    return res;
}
int main(){
    n=read(),k=read();
    F(i,1,n) a[i]=read();
    coef[0]=1;
    F(i,1,n) DF(j,n,0) coef[j]=(coef[j-1]+1ll*coef[j]*a[i])%P;
    int mul=1,ans=0;
    F(i,0,min(n,k)){
        ans=(ans+1ll*mul*qmi(n,k-i)%P*coef[i])%P;
        mul=1ll*mul*(k-i)%P;
    }
    ans=1ll*ans*qmi(qmi(n,k),P-2)%P;
    printf("%d\n",ans);
    fprintf(stderr,"%d ms\n",int(1e3*clock()/CLOCKS_PER_SEC));
    return 0;
}