西南交大峨眉校区INT杯2020-K.建教室【树的直径+LCA倍增】

建教室

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;
}