【五校联考2015 9.5】送你一棵线段树

题目大意

你有$n$棵树。每一棵树的形态如下给出：

• 0号树$T_0$只有一个点，标号为0
  • 对于第$i$棵树$T_i(i>0)$，它通过拷贝第$A_i$和$B_i$棵树，并将对应原树上的$C_i$号和$D_i$号节点链接一条$L_i$的边得到。并且新树上原本属于$B_i$节点的节点标号都加上了$size_{A_i}$。

  现在要求你计算每一棵树的这个值

  $\sum_{i\in T_i}\sum_{j\in T_i}dis(i, j)$

  其中$dis(i, j)$表示树上点$i$和点$j$之间的最短路的长度。

  $n<60, A_i < i, B_i < i, Case<100$

  ---

  分析

  不难发现最坏情况下，$size_{T_n}$是可以达到$2^n$的。
  我们先想一下如果我们要计算一棵树的答案需要知道些什么，很自然可以想到从这棵树的构成出发。它等于$A_i$的答案加上$B_i$的答案再加上跨过$A_i$和$B_i$的所有路径的长度，前两部分可以递归地算，主要是看后一部分。

  后一部分显然等于

  $size_{A_i}sum_{D_i}+size_{B_i}sum_{C_i}+L_isize_{A_i}size_{B_i}$

  $size$数组很好统计，主要是看如何计算$sum$。
  考虑到每棵树上直接需要查询$sum$值的点顶多只有$n$个，间接需要查询（即后面的树上的某个点属于前面某棵树上的点，我们依旧需要把它算出）的点顶多只有$n^2$个。
  那么依照前面的思路，我们拆成两棵树来分别维护这些关键点的$sum$值，假设$x$是$T_i$的关键点且$x\in T_{A_i}$（$x\in T_{B_i}$是同理的）。

  $sum_{T_i, x}=sum_{A_i, x}+[dis(C_i, x)+L_i]size_{B_i}$

  那么新的问题就出现了，我们还有维护关键点之间的$dis$值。其实思路是同理的，就略推导的式子了。

  最终的时间复杂度$O(n^3Case)$