P2195 HXY造公园 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)
树的直径
题意:操作1查询某点所在树的直径。操作2合并两棵树,使得合并后的树的直径最短。
操作1:可用树上的DP求解。值得注意的是,本题需要多次查找树的直径,可使用某点所在树的根节点的直径表示,记为数组c[x]。
操作2:将直径中心点合并会使得合并后树的直径最小。因为合并后的树的直径一定大于等于原树的直径,如果存在大于原树的直径的链,那么这条链一定过合并的那条边通向原本另一个树。这时只需让这条链在原树的部分的最大值最小,故链的起点一定在直径的中点。(在直径不在中点一定比中点长,不在直径上到直径还需一段距离也比中点长)。合并后只需使用路径压缩法,使树A的根的父亲指向树B的根节点,树B的根节点的直径数组改为max{两棵树直径的一半(向上取整)的和+1,原本两棵树的直径}即可,此时无论查找树A还是树B都可以返回合并后的树的直径。值得注意的是,本题不需要关心两棵树具体是哪两个点进行的合并,我们只需要关心合并后树的直径和各点的集合关系即可。