[ 生成函数 ] Codeforces891E Lust

设 $b_i$ 表示 $a_i$ 减了几次，那么答案就是 $\prod_{i=1}^na_i-\prod_{i=1}^n(a_i-b_i)$ 。可以对每次操作分开计算贡献得到。
然后就是怎么求 $\prod_{i=1}^n(a_i-b_i)$ 的期望。令 $k=\sum{b_i}$ 。

$$E={1\over n^k}{k!\over {\prod_{i=1}^n b_i!}}\prod_{i=1}^n{(a_i-b_i)}={k!\over n^k}·\prod_{i=1}^n{(a_i-b_i)\over {b_i!}}$$
然后考虑怎么求 $\prod_{i=1}^n{(a_i-b_i)\over {b_i!}}$ 。
构造生成函数：
$$F(x)=\prod_{i=1}^n\sum_{j=0}^{\infty}{a_i-j\over{j!}}x^j=\prod_{i=1}^n\sum_{j=0}^{\infty}{a_ix^j\over{j!}}-{x·x^{j-1}\over (j-1)!}=\prod_{i=1}^n (a_i-x)e^x=e^{nx}\prod_{i=1}^n (a_i-x)$$
令 $G(x)=\prod_{i=1}^n(a_i-x)=\sum_{i=0}^nc_ix^i$ 。
其中 $c_i$ 可以 $O(n^2)$ 求出。
$$[x^k]F(x)=\sum_{i=0}^nc_i{n^{k-i}\over (k-i)!}$$
$$E=\sum_{i=0}^nc_i{\prod_{j=k-i+1}^k j\over n^i}$$
然后就好了。

#include<bits/stdc++.h>
using namespace std;
const int N=5010;
const int M=1000000007;
int k,n,m,x,Ans,s=1;
int c[N],d[N];
inline void Add(int& x,int y){
    x=(x+y)%M;
}
inline int Pow(int x,int y){
    int Ans=1;
    for(;y;y>>=1,x=1ll*x*x%M)
    if(y&1)Ans=1ll*Ans*x%M;
    return Ans;
}
int main(){
    scanf("%d%d",&n,&k);
    c[0]=1;
    for(int i=1;i<=n;i++){
        scanf("%d",&x);
        s=1ll*s*x%M;
        for(int j=0;j<i;j++)d[j]=1ll*c[j]*x%M;
        for(int j=0;j<i;j++)Add(d[j+1],-c[j]);
        memcpy(c,d,sizeof(d));
    }
    int inv=Pow(n,M-2);
    for(int i=0,t=1,T=1;i<=n;i++){
        Add(Ans,1ll*c[i]*T%M*t%M);
        t=1ll*t*inv%M;T=1ll*T*(k-i)%M;
    }
    Add(s,-Ans);
    cout<<(s+M)%M<<endl;
    return 0;
}