思路:
这道题问树上任意两点距离,我们很自然地想到LCA求距离。
树的直径有一个性质:
现在有两棵树,如果把它们随意连一条边,会变成一棵树,新树的直径的端点一定是之前两棵树的直径的共4个端点的两个。
即我们设(x1,x2)表示一开始那棵树的直径两端,(y1,y2)表示新增两个叶子节点后的直径两端。
最后答案只有在x1~y1, x2y1,x1x2三条距离取一个最长的,即是新增点后的书的直径。
(因为y1,y2的深度是一样的,所以我们只用随便枚举一点作为树直径的一端)
#include <cstdio>
#include <iostream>
using namespace std;
const int N = 1e5 + 10;
int m, x, f[N << 1][21], som[N << 1], deep[N << 1], ans, num = 4, s = 2, t =4;
int LCA(int x, int y)
{
if(deep[x] < deep[y]) swap(x, y);
for(int i = 20; i >= 0; i--)
if(deep[f[x][i]] >= deep[y])
x = f[x][i];
if(x == y) return x;
for(int i = 20; i >= 0; i--)
if(f[x][i] ^ f[y][i])
x = f[x][i], y = f[y][i];
return f[x][0];
}
int Dis(int x, int y)
{
int res = LCA(x, y);
return deep[x] + deep[y] - (deep[res] << 1);
}
void add(int now)
{
deep[++num] = deep[now] + 1;
f[num][0] = now;
for(int i = 1; i <= 20; i++)
f[num][i] = f[f[num][i - 1]][i - 1];
}
int main()
{
f[2][0] = f[3][0] = f[4][0] = 1;
deep[2] = deep[3] = deep[4] = 1;
// deep[1] = 1;
scanf("%d", &m);
for(int i = 1; i <= m; i++)
{
scanf("%d", &x);
add(x), add(x);
int dis = Dis(s, t), dis2, dis3;
dis2 = Dis(s, num);
dis3 = Dis(t, num);
if(dis3 >= dis && dis3 >= dis2)
s = num, printf("%d\n", dis3), dis = dis3;
else if(dis2 >= dis && dis2 >= dis3)
t = num, printf("%d\n", dis2), dis = dis2;
else printf("%d\n", dis);
}
return 0;
}