JZOJ 4888. 【NOIP2016提高A组集训第14场11.12】最近公共祖先

JZOJ 4888. 【NOIP2016提高A组集训第14场11.12】最近公共祖先

注意到每一层的结点作为 lca 的贡献是相同的。

设当前的结点的深度为 $d$， 深度为 $d$ 的子树大小为 $s{z}_d$。

那么这个结点的贡献是 $s{z}_{d+1}^{2}k(k-1)+2s{z}_d-1$。

比较显然的 $s{z}_d={\sum }_{i=d}^{n-1}{k}^{i-d}$，通过等比数列求和得 $s{z}_d=\frac{k^{n-d}-1}{k-1}$。

当然答案是 ${\sum }_{d=0}^{n-1}[s{z}_{d+1}^{2}k(k-1)+2s{z}_d-1]\times d\times k^d$。

化简一下
$$\begin{array}{rl}& \sum _{d=0}^{n-1}[s{z}_{d+1}^{2}k(k-1)+2s{z}_d-1]\times d\times k^d\\ =& \sum _{d=0}^{n-1}[(\frac{k^{n-d-1}-1}{k-1}{)}^{2}k(k-1)+2\frac{k^{n-d}-1}{k-1}-1]\times d\times k^d\\ =& \frac{1}{k-1}\sum _{d=0}^{n-1}[(k^{n-d-1}-1{)}^{2}k+2(k^{n-d}-1)-(k-1)]\times d\times k^d\\ =& \frac{1}{k-1}\sum _{d=0}^{n-1}[(k^{2n-2d-2}-2k^{n-d-1}+1)k+2k^{n-d}-2-k+1]\times d\times k^d\\ =& \frac{1}{k-1}\sum _{d=0}^{n-1}(k^{2n-2d-1}-2k^{n-d}+k+2k^{n-d}-2-k+1)\times d\times k^d\\ =& \frac{1}{k-1}\sum _{d=0}^{n-1}(k^{2n-d-1}-k^d)\times d\end{array}$$
目标，计算${\sum }_{d=0}^{n-1}{k}^d\times d$。

看似不可求解。实则可以转化。
$$\begin{array}{rl}& \sum _{d=0}^{n-1}{k}^d\times d\\ =& \sum _{d=0}^{n-1}\sum _{d^{\prime }=0}^d{k}^{d^{\prime }}\\ =& \sum _{d=0}^{n-1}\frac{k^{d+1}-1}{k-1}\\ =& \frac{1}{k-1}(\sum _{d=0}^{n-1}{k}^{d+1}-n)\\ =& \frac{1}{k-1}(\frac{k^{n+1}-k}{k-1}-n)\end{array}$$
把这个代入进去，就可以 $O(\log p)$ 求解。

题解给了一个简洁的式子，我这里就不再推导了。