jzoj 5898. 【NOIP2018模拟10.6】距离统计 动态树分治

Description
Input

Output

Sample Input

5 3
1 2 3
1 3 1
2 4 4
2 5 2
1 2
3 3
5 1

Sample Output

3
6
2

Data Constraint

分析：
我们可以先二分一个答案，那么问题就变成了从一个点$x$开始的路径长度$\le mid$的有多少条。
考虑用动态树分治。
每个点维护一个vector表示以这个点开始的路径的长度，并排序。还维护一些vector表示他的每一棵子树的到他路径上的长度，并排序。因为从$x$开始，那么$x$所在子树是没有贡献的，可以用总贡献减去这棵子树的贡献。然后再维护一个map表示每个点属于第几棵子树。
距离可以用倍增lca求。
总复杂度是$O(nlo{g}^{3}n)$的。
好像数据有锅，所有人都是85分。

代码：

#include 
#include 
#include 
#include 
#include