CodeForces-1304E 1ツリーとクエリlcaはツリー上の2点間の距離を検出します

CodeForces-1304E 1ツリーとクエリ

元のタイトルの住所：

http://codeforces.com/problemset/problem/1304/E

トピック：ツリーを与え、qクエリを実行し、クエリを実行するたびにツリーにエッジを追加して、この時点での2点間の距離が特定の値kになるかどうかを尋ねます（同じノードを繰り返し通過できます）または同じ側）。

基本的な考え方：xとyを接続するエッジを追加した後、aとbの間の距離には、dist（a、b）、dist（a、x）+ dist（y、b）+1の3つの形式があります。 、dist（a、y）+ dist（x、b）+1;次に、同じノードまたは同じエッジを繰り返し通過できるため、答えは上記の3つの可能性+ 2 * k（kは任意の自然数）でなければなりません。直接ループの判断は問題ありません。奇数と偶数のフォームを判断するためにここにいます（直接ループほど良くありません）。
知識の主な困難は、ツリー上の2点間の距離をlcaを介してすばやく見つけることであり、テンプレートを直接設定できます。

実装コード：

#include <bits/stdc++.h>
using namespace std;
#define IO std::ios::sync_with_stdio(false)
#define int long long
#define INF 0x3f3f3f3f
typedef unsigned int uint;

const int maxn = 1e5 + 10;
int n,q;
struct edge {
    int next, v;
}edges[maxn*2];
int cnt;
int head[maxn];
void init() {
    memset(head, -1, sizeof(head));
    cnt = 0;
}
void add_edge(int u,int v) {
    edges[cnt].next = head[u];
    edges[cnt].v = v;
    head[u] = cnt++;
}
int dep[maxn];
int f[maxn][21];
void dfs(int u,int fa) {
    dep[u] = dep[fa] + 1;
    for (int i = 0; i <= 19; i++)
        f[u][i + 1] = f[f[u][i]][i];
    for (int i = head[u]; i != -1; i = edges[i].next) {
        int v = edges[i].v;
        if (v == fa)
            continue;
        f[v][0] = u;
        dfs(v, u);
    }
}
int lca(int x,int y) {
    if (dep[x] < dep[y])
        swap(x, y);
    for (int i = 20; i >= 0; i--) {
        if (dep[f[x][i]] >= dep[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 dist(int a,int b){
    return dep[a] + dep[b] - 2 * dep[lca(a, b)];
}
signed main() {
    IO;
    cin >> n;
    init();
    for (int i = 1; i <= n - 1; i++) {
        int u, v;
        cin >> u >> v;
        add_edge(u, v);
        add_edge(v, u);
    }
    dfs(1, 0);
    cin >> q;
    set<int> memo;
    while (q--) {
        int x, y, a, b, k;
        memo.clear();
        cin >> x >> y >> a >> b >> k;
        memo.insert(dist(a, b));
        memo.insert(dist(a, x) + dist(y, b) + 1);
        memo.insert(dist(a, y) + dist(x, b) + 1);
        int f1 = INF, f2 = INF;//偶数的最小值，奇数的最小值;
        for (auto m : memo) {
            if (m % 2 == 0) f1 = min(m, f1);
            else f2 = min(m, f2);
        }
        //大于最小偶数的偶数，大于最小奇数的奇数;
        if ((k % 2 == 0 && k >= f1) || (k % 2 == 1 && k >= f2)) {
            cout << "YES" << endl;
        } else cout << "NO" << endl;
    }
    return 0;
}