看到大佬们都是用的 O ( n ) O(n) O(n) 的做法,本蒟蒻不会,就按合情合理的思路写了一个 O ( n log n ) O(n\log n) O(nlogn) 的做法。
首先可以确定,其中的两个点肯定是直径的两个端点。感性理解一下,如果不是的话,那么把其中的点换成直径的一个端点,答案一定不会更差,因为直径已经是树上的最长路径了。然后问题就变成了在树上找到一个点,使它到直径的两个端点的路径和直径的并所经过的边数最多。我们考虑一个简单的不能算容斥的容斥。
拿这棵树来说,假如直径的两个端点是 1 1 1 和 6 6 6,我们要找 5 5 5 这个点,那么分别记录下来直径的两个端点到其余点的距离,然后我们将其中的一个端点设为根,把其余点的深度和求 l c a lca lca 用的数组预处理出来。那么我们可以求出 5 5 5 和深度较深的那个端点的 l c a lca lca,然后就是一个小的容斥,假设两个端点到 5 5 5 的距离分别为 d i s 1 dis1 dis1 和 d i s 2 dis2 dis2,那么答案就是 d i s 1 + d i s 2 + d e p [ 5 ] − d e p [ l c a ] dis1+dis2+dep[5]-dep[lca] dis1+dis2+d