[洛谷P4707]重返现世

Description

有n种物品，每次操作你有$\frac{pi}{m}$的概率获得第i种物品
求你获得k种物品的期望操作数。
n<=1000,n-k<=10,m<=10000

Solution

考虑第i个物品的EGF为$e^{\frac{p_i}{m}x}-1$
枚举最后一个出现的物品为s，令$$F_s(x)=\sum _{|a|=k-1,s\notin a}\prod _{j=1}^{k-1}(e^{\frac{p_j}{m}x}-1)$$
我们要求的就是$$\frac{p_s}{m}\sum _{i>=0}[x^i]F_s(x)i!\ast i$$
考虑把F(x)写成$$\sum a_i{e}^{\frac{b_i}{m}x}$$
对于一个$e^{\frac{n}{m}x}$对于答案的贡献为$$\sum _{i>=0}\frac{(\frac{n}{m}{)}^i}{i!}\ast i!\ast (i+1)$$
$$=\sum _{i>=0}(\frac{n}{m}{)}^i(i+1)$$
$$=(\frac{m}{m-n}{)}^2$$
考虑如何求F(x)，先不管i不能选的限制，求出F[k][j]表示当前有k个多项式没有选择，剩下的多项式的乘积
然后每一次对于枚举的i，从F[k][j]中将i的贡献删去，统计答案之后再加回来
复杂度O(n(n-k)m)

Code

#include <cstdio>
#include <cstring>
#include <algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
using namespace std;

typedef long long ll;

const int N=1e4+5,Mo=998244353;

int inc(int x,int y) {return x+y>=Mo?x+y-Mo:x+y;}
int dec(int x,int y) {return x-y<0?x-y+Mo:x-y;}

int n,m,k,p[N],f[15][N],inv[N];

int main() {
	scanf("%d%d%d",&n,&k,&m);k=n-k;
	fo(i,1,n) scanf("%d",&p[i]);
	inv[1]=1;fo(i,2,m) inv[i]=(ll)(Mo-Mo/i)*inv[Mo%i]%Mo;
	f[0][0]=1;
	fo(i,1,n) {
		fd(w,k,0)
			fd(j,m,0) {
				f[w][j]=dec(0,f[w][j]);
				if (j>=p[i]) f[w][j]=inc(f[w][j],f[w][j-p[i]]);
				if (w) f[w][j]=inc(f[w][j],f[w-1][j]);
			}
	}
	int ans=0;
	fo(i,1,n) {
		fo(j,0,m) 
			fo(w,0,k) {
				if (w) f[w][j]=dec(f[w][j],f[w-1][j]);
				if (j>=p[i]) f[w][j]=dec(f[w][j],f[w][j-p[i]]);
				f[w][j]=dec(0,f[w][j]);
			}
		int res=f[k][0];
		fo(j,1,m-p[i]) {
			int p=(ll)m*inv[m-j]%Mo;
			res=inc(res,(ll)f[k][j]*p%Mo*p%Mo);
		}
		ans=inc(ans,(ll)res*p[i]%Mo*inv[m]%Mo);
		fd(w,k,0)
			fd(j,m,0) {
				f[w][j]=dec(0,f[w][j]);
				if (j>=p[i]) f[w][j]=inc(f[w][j],f[w][j-p[i]]);
				if (w) f[w][j]=inc(f[w][j],f[w-1][j]);
			}
	}
	printf("%d\n",ans);
	return 0;
}