建教室
Description
渣哥所在的学校最近建起了新教室。每当学校建起一个新教室的时候,学校会将这个教室用一条双向通行道路与一个现有教室连接起来。因为渣哥非常热爱运动,他喜欢在教室间穿梭。当一个新教室被建造,他想要知道从某个教室出发,到距离这个教室最远的教室的距离。两个教室间的距离为这两个教室间的道路的条数。
接下来会给出总共N次指令,每个指令都是“建造”和“求距离”之一。对于一个建造指令,学校建造一个教室,并将其与至多一个现有的教室连接起来。对于一个距离指令,渣哥向你询问从某个特定的教室到离它最远的教室的距离。保证询问的教室已经被建造。
Input
第一行一个整数N。第2行到第N+1行每行一个大写字符“Q”或者“B”以及一个整数x。若为Q,你需要输出教室x能到达的最远的教室的距离。若为B,则表示新建一个教室,并且按编号从小到大顺序建造(第一次建造1号教室,第二次建造2号教室,以此类推),这个教室会建造一条双向通行道路连向x。若x为-1,则表示这个教室不被连向任何现有教室。N≤ 100000
Output
对于每个距离指令,输出上面定义的最长距离。
思路:
前置知识:在一个树中,最远点对称为树的直径,如果某个点走一条路能到达的最远距离,必然是树直径的某个端点。所以我们需要用到一个f数组来存每个结点的根节点,用一个结构体存每个根节点的左右端点。然后利用倍增LCA来求两点之间的距离(deep[x]+deep[y] - 2*deep[LCA(x,y)]); 再利用LCA来更新祖先关系。
更新操作:因为最长距离只跟树的左端点和右端点有关,所以在插入新的结点时,需要判断是否会影响到两个端点,如果影响则更新。
查询操作:直接求x与该根节点两个端点的距离,去最大值。
AC代码:
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int MAXN = 1e5 + 5;
int f[MAXN], fa[MAXN][20], deep[MAXN];
struct Node{
int l, r;
}node[MAXN];
int LCA(int x, int y) {
if (deep[x] < deep[y]) {
swap(x, y);
}
// 先将x跳到y层
for (int i = 19; i >= 0; --i) {
if ((deep[x] - deep[y]) >> i) {
x = fa[x][i];
}
}
if (x == y) return x;
for (int i = 19; i >= 0; --i) {
if (fa[x][i] != fa[y][i]) {
x = fa[x][i];
y = fa[y][i];
}
}
return fa[x][0];
}
int dis(int x, int y) {
return deep[x] + deep[y] - 2 * deep[LCA(x, y)];
}
void solve() {
int cnt = 0, t, x, rt;
scanf("%d",&t);
char ch;
while(t--) {
scanf(" %c%d", &ch, &x);
if (ch == 'Q') {
rt = f[x];
//计算该结点到树直径的两个端点,输出较远的那条
printf("%d\n", max(dis(x, node[rt].l), dis(x, node[rt].r)));
continue;
}
cnt++;
if (x == -1) {
f[cnt] = cnt;
deep[cnt] = 1;
node[cnt].l = node[cnt].r = cnt;
} else {
//将添加的新结点更新
rt = f[x];
f[cnt] = rt;
fa[cnt][0] = x;
deep[cnt] = deep[x] + 1;
//更新该节点的所有祖先关系
for (int i = 1; i < 20; ++i) {
fa[cnt][i] = fa[fa[cnt][i - 1]][i - 1];
}
int d = dis(node[rt].l, node[rt].r), dl = dis(node[rt].l, cnt), dr = dis(node[rt].r, cnt);
if (dl > d) {
node[rt].r = cnt;
continue;
}
if (dr > d) {
node[rt].l = cnt;
}
}
}
}
int main() {
solve();
return 0;
}