把只能选没死亡敌人的限制去掉,改成每次随机选一个攻击,选到死了的就不计次数,每次概率乘的是 $\frac{1}{n}$。枚举最后一次有效的攻击的敌人 $last$,并用指数型生成函数 $F(x,y)$ 表示一共打 $x$ 次,有效次数 $y$ 次的概率和。
令 $t=\frac{x}{n}$,则被击杀的函数为 $A_p=y^{a_p}(\exp t-\sum_{i=0}^{a_p-1}\frac{t^i}{i!})$,不被击杀的函数为 $B_p=\sum_{i=0}^{a_p-1}\frac{t^iy^i}{i!}$。
讨论每个位置的贡献函数,最终乘起来得到多项式 $G$,需要求的形如 $\sum i!\times [x^iy^{k-1}]G$。问题在于 $\exp t$ 会导致项数是无穷的,考虑直接令 $z=\exp t$ 并记录选中 $z$ 的次数,最终多项式形如 $f_{a,b,c} t^ay^bz^c$。
考虑 $H(a,c)=t^a\exp(ct)=\frac{1}{n^a}x^a\exp(\frac{c}{n}x)$,需要求 $w\sum[x^i]i!(x^a\exp(bx))$。考虑 $\sum[x^i]i!(x^a\exp(bx))$ 最终即为 $F=\sum_{i\ge 0}\frac{x^{a+i}(a+i)!b^i}{i!}$ 系数和(或者 $x=1$ 的点值),转化为 $a!x^a\sum(bx)^i\binom{i+a}{i}=a!x^a\frac{1}{(1-bx)^{a+1}}$。最终贡献是 $\frac{a!}{(1-b)^{a+1}}$。
求 $G(last)$ 支持多项式除法,求出全局的结果后重新算这一项的贡献即可,时间复杂度 $\mathcal O(n^4V^3)$。