codeforces 891E Lust 题解

题目传送门

题目大意： 给出一个长度为 $n$ 的序列，进行 $k$ 次操作，每次随机选取一个 $a_i$，让答案加上 $a$ 序列中的其他数的乘积，然后让 $a_i-1$，问最后答案的期望。

题解

膜拜往日的神仙 OZY。

与样例快乐van耍之后，发现答案与操作顺序无关，只与每个位置被操作的次数有关。

那么不妨认为每个数都是被连续操作的，那么我们可以把一次操作的贡献看成 $(a_x\prod _{i\ne x}{a}_i)-((a_x-1)\prod _{i\ne x}{a}_i)$，那么把 $s$ 次对 $x$ 的连续操作加起来，就得到了 $(a_x\prod _{i\ne x}{a}_i)-((a_x-s)\prod _{i\ne x}{a}_i)$。

同样的，把 $k$ 次操作后的贡献加起来，就能得到 $\prod a_i-\prod (a_i-b_i)$，其中 $b_i$ 是第 $i$ 个数被操作的次数。

前面一项是固定的，只需要求后面一项的期望即可。

对每一个数 $a_j$ 造一个生成函数，$x^i$ 项表示这个数被操作 $i$ 次，系数就是被操作 $i$ 次的贡献，即 $(a_j-i)$。

考虑到所有操作形成一个排列，一个 $b$ 序列可能对应多种操作方式，所以用 $EGF$，有 $f_j=\sum _{i=0}(a_j-i)\frac{x^i}{i!}$。

那么将所有 $f_j$ 乘起来，$x^k$ 那一项的系数就是答案。但是 $x^k$ 下面有个 $k!$，系数乘过来的时候得到的是 $x^k$ 的系数，而不是 $\frac{x^k}{k!}$ 的系数，所以这一项的系数还要乘以 $k!$，才是我们想要的 $\frac{x^k}{k!}$ 的系数。

并且，由于需要求的是期望，而这一项的系数是所有方案的和，所以还要除以总方案数 $n^k$。

所以，最终的柿子为：
$$\begin{array}{rl}& \frac{k!}{n^k}\prod _{j=1}^n(\sum _{i=0}(a_j-i)\frac{x^i}{i!})\\ & =\frac{k!}{n^k}\prod _{j=1}^n(\sum _{i=0}{a}_j\frac{x^i}{i!}-\sum _{i=0}i\frac{x^i}{i!})\\ & =\frac{k!}{n^k}\prod _{j=1}^n(a_j{e}^x-\sum _{i=0}\frac{x^i}{(i-1)!})\\ & =\frac{k!}{n^k}\prod _{j=1}^n(a_j{e}^x-\sum _{i=0}x\frac{x^i}{i!})\\ & =\frac{k!}{n^k}\prod _{j=1}^n(a_j-x)e^x\\ & =\frac{k!}{n^k}{e}^{nx}\prod _{j=1}^n(a_j-x)\end{array}$$

后面的 $\prod$ 可以 $n^2$ 暴力求，设 $f[i]$ 表示求完之后的多项式的 $x^i$ 项的系数。

由于需要求第 $k$ 项的系数，而后面的 $\prod$ 得到的是个 $n$ 次的多项式，所以只需要 $e^{nx}$ 后 $n$ 项的系数。

最后的答案就是：
$$\begin{array}{rl}ans& =\frac{k!}{n^k}\sum _{i=0}^{\min (n,k)}f[i]\times \frac{n^{k-i}}{(k-i)!}\\ & =\sum _{i=0}^{\min (n,k)}f[i]\times \frac{(k-i+1)(k-i+2)...k}{n^i}\end{array}$$

代码如下：

#include <cstdio>
#define mod 1000000007

int n,k,f[5010],ans=0;
int ksm(int x,int y){int re=1;for(;(y&1?re=1ll*re*x%mod:0),y;y>>=1,x=1ll*x*x%mod);return re;}
inline int min(int x,int y){return x<y?x:y;}
#define inv(x) ksm(x,mod-2)

int main()
{
	scanf("%d %d",&n,&k);f[0]=1;
	for(int i=1,x;i<=n;i++){
		scanf("%d",&x);for(int j=i;j>0;j--)
		f[j]=(1ll*f[j]*x%mod-f[j-1]+mod)%mod;f[0]=1ll*f[0]*x%mod;
	}
	for(int i=0,fac=1,n_=1;i<=min(n,k);i++){
		ans=(ans+1ll*f[i]*fac%mod*inv(n_)%mod)%mod;
		fac=1ll*fac*(k-i)%mod;n_=1ll*n_*n%mod;
	}
	printf("%d",(f[0]-ans+mod)%mod);
}