需要注意题目求解的是操作 n 次出现 x 次王牌,而不是第一次出现王牌是第 x 次。
令 p=m1 也就选出王牌的概率,q=1−p。
那么如果直接枚举出现王牌的数量,有式子:
i=0∑n(in)piqn−iik
根据套路把 ik 展开为第二类斯特林数:
i=0∑n(in)piqn−iik=i=0∑n(in)piqn−ij=1∑k{kj}(i−j)!i!
因为我们希望最后的时间复杂度与 n 无关,先把与 n 有关的式子移动到外面:
i=0∑n(in)piqn−iik=j=1∑k{kj}i=0∑n(in)piqn−i(i−j)!i!
考虑把后面的一坨式子化简,阶乘一般的处理手段都是加入组合数然后根据组合数的性质快速计算。
i=0∑n(in)piqn−i(i−j)!i!=i=0∑ni!(n−i)!n!×(i−j)!i!piqn−i
因为阶乘很狗屎,所以先处理阶乘。
i=0∑n(i−jn−i)(n−j)!n!piqn−i
把式子分开看:
(n−j)!n!i=0∑n−j(in−i)pi+j(1−p)n−i−j
这里需要补一下二项式定理:
(a+b)n=i=0∑n(in)aibn−i
发现跟上面的式子有点像,考虑把 pi 提出来,得到:
(n−j)!n!pj(p+(1−p))n−i=(n−j)!n!×pj
将化简之后的结果带回原式,得到:
j=0∑k{kj}(n−j)!n!pj
虽然是垃圾的 O(klogn),但是你就说能不能过吧。