[CF891E]Lust

题目大意

一个序列a，做k次下列操作：
1、随机一个下标x，答案加上$\Pi_{i=1,i!=x}^na_i$
2、将$a_x$减一。
求答案的期望。

做法

设$b_i$表示最终$a_i$减少了多少次。
答案是$\Pi_{i=1}^na_i-\Pi_{i=1}^n(a_i-b_i)$
$E(\Pi_{i=1}^na_i-\Pi_{i=1}^n(a_i-b_i))=\Pi_{i=1}^na_i-\frac{k!\sum_{\sum_{i=1}^nb_i=k}\Pi_{i=1}^n\frac{(a_i-b_i)}{b_i!}}{n^k}$
只考虑计算后面部分。
设$F(x)=\Pi_{i=1}^n\sum_{j>=0}\frac{x^j*(a_i-j)}{j!}$
这是后面那个东西的生成函数。
$F(x)=\Pi_{i=1}^ne^x(a_i-x)$
$F(x)=e^{nx}\Pi_{i=1}^n(a_i-x)$
对于后面部分我们暴力展开得到$\sum_{i=0}^nc_ix^i$
现在$[x^k]F(x)=\sum_{i=0}^nc_i\frac{n^{k-i}}{(k-i)!}$
代入原式得到$\sum_{i=0}^nc_i\frac{\Pi_{j=k-i+1}^kj}{n^i}$

#include<cstdio>
#include<algorithm>
#define fo(i,a,b) for(i=a;i<=b;i++)
using namespace std;
typedef long long ll;
const int maxn=5000+10,mo=1000000007;
int c[maxn],d[maxn],a[maxn];
int i,j,k,l,t,n,m,ni,num,ans;
int qsm(int x,int y){
    if (!y) return 1;
    int t=qsm(x,y/2);
    t=(ll)t*t%mo;
    if (y%2) t=(ll)t*x%mo;
    return t;
}
int main(){
    scanf("%d%d",&n,&k);
    c[0]=1;
    fo(i,1,n){
        scanf("%d",&a[i]);
        fo(j,0,i) d[j]=0;
        fo(j,0,i-1){
            (d[j]+=(ll)c[j]*a[i]%mo)%=mo;
            (d[j+1]-=c[j])%=mo;
        }
        fo(j,0,i) c[j]=d[j];
    }
    ni=qsm(n,mo-2);
    num=1;
    fo(i,0,n){
        (ans+=(ll)c[i]*num%mo)%=mo;
        num=(ll)num*ni%mo*(k-i)%mo;
    }
    ans=(c[0]-ans)%mo;
    (ans+=mo)%=mo;
    printf("%d\n",ans);
}