两个无限项求和问题（生成函数）

对 $k=0\sim K$ 求 $\sum _{i=0}^{\infty }{p}^i\left(\genfrac{}{}{0ex}{}{i}{k}\right)$。

$$\begin{array}{rl}& \sum _{i=0}^{\infty }{p}^i\left(\genfrac{}{}{0ex}{}{i}{k}\right)\\ =& \sum _{i=0}^{\infty }{p}^i[x^k](1+x{)}^i\\ =& [x^k]\sum _{i=0}^{\infty }(p(1+x){)}^i\\ =& [x^k]\frac{1}{1-p-px}\end{array}$$

于是可以 $O(K\log K)$ 求了。

哦不好意思上面都是在瞎扯（

实际上：
$$\begin{array}{rl}& \sum _{i=0}^{\infty }{p}^i\left(\genfrac{}{}{0ex}{}{i}{k}\right)\\ =& \frac{1}{k!}\sum _{i=k}^{\infty }{p}^i{i}^{\underline{k}}\end{array}$$
注意到：
$$\begin{array}{r}{\left(\frac{1}{1-x}\right)}^{(k)}={\left(\sum _{i=0}^{\infty }{x}^i\right)}^{(k)}=\sum _{i=k}^{\infty }{i}^{\underline{k}}{x}^{i-k}\end{array}$$
而由数学归纳法可知：
$${\left(\frac{1}{1-x}\right)}^{(k)}=\frac{k!}{(1-x{)}^{k+1}}$$
那么 $x^k\frac{k!}{(1-x{)}^{k+1}}=\sum _{i=k}^{\infty }{i}^{\underline{k}}{x}^i$。

于是：
$$\begin{array}{rl}& \sum _{i=0}^{\infty }{p}^i\left(\genfrac{}{}{0ex}{}{i}{k}\right)\\ =& \frac{1}{k!}\sum _{i=k}^{\infty }{p}^i{i}^{\underline{k}}\\ =& \frac{1}{k!}{p}^k\frac{k!}{(1-p{)}^{k+1}}\end{array}$$
于是 $O(K)$ 预处理好阶乘之后单个 $p,k$ 就可以 $O(1)$ 求了。

对 $k=0\sim K$ 求 $\sum _{i=0}^{\infty }{p}^i{i}^k$。

$$\begin{array}{rl}& \sum _{i=0}^{\infty }{p}^i{i}^k\\ =& \sum _{i=0}^{\infty }{p}^{i}k![x^k]e^{ix}\\ =& k![x^k]\sum _{i=0}^{\infty }(p{e}^x{)}^i\\ =& k![x^k]\frac{1}{1-p{e}^x}\end{array}$$

于是可以 $O(K\log K)$ 求了。

事实上你也可以用第二类斯特林数展开 $i^k$，然后套用第一个问题的公式得到：
$$\sum _{i=0}^{\infty }{p}^i{i}^k=\sum _{l=0}^k\left\{\begin{array}{c}k\\ l\end{array}\right\}{p}^l\frac{l!}{(1-p{)}^{l+1}}$$
这样单个 $p,k$ 是 $O(K)$ 的，如果要求多个 $k$ 的话就是 $O(K^2)$ 的了。

所以说这次 $O(K\log K)$ 是正经的。