显然,应该要加入 ( 1 , v ) (1,v) (1,v) 的边。
假设对于加的边为 x x x ,最后答案为 a n s ans ans ,那么对于小于 x x x 的答案不会超过 a n s ans ans 。
我们可以考虑如果要让答案为 a n s ans ans , x x x 那条边要怎么加。
我们只关心 d e p > a n s dep>ans dep>ans 的点。
因此我们希望走过 x x x 这条边后,走到这些点的路程最短。
类似树的直径,我们需要知道这些点中距离最远的点对的距离是多少,我们把它记为 f a n s f_{ans} fans 。
因此我们只需要知道 f a n s f_{ans} fans ,然后对于 x x x 和 a n s ans ans 做个双指针即可。
那么 f a n s f_{ans} fans 怎么得到呢?
对于每个点 v v v ,我们只关心它不同子树的最长、次长链,假设为 x 1 , x 2 x1,x2 x1,x2 ,那么 f i < x 2 = max ( f i , x 1 + x 2 − 2 ∗ d e p v ) f_{i<x2}=\max(f_{i},x1+x2-2*dep_v) fi<x2=max(fi,x1+x2−2∗depv) , f i < d e p v = max ( f i , x 1 − d e p v ) f_{i<dep_v}=\max(f_{i},x1-dep_v) fi<depv=max(fi,x1−depv) 。
因此记录在 f x 2 − 1 f_{x2-1} fx2−1 和 f d e p v − 1 f_{dep_v-1} fdepv−1 ,最后做个后缀 max \max max 即可。
效率: O ( n ) O(n) O(n) 。