Codeforces Round 620 (Div. 2) E. 1-Trees and Queries（LCA）

原题链接：E. 1-Trees and Queries

---

题目大意:

---

给定一棵 $n$ 个点，$n-1$ 条边的树，规定树边的边权为 $1$ 。

现在有 $q$ 次询问，每次询问给出五个正整数：$x,y,a,b,k$ 。

即在原图 $(x,y)$ 两点间连上一条特殊的新边，边权也为 $1$（保证加边后原图中不会出现重边和自环），你要判断从点 $a$ 出发到达点 $b$ 是否有一条长度为 $k$ 的路径。

注意，每次的询问是独立的，即这一次询问加上的边不能为下一次询问所使用，且该路径是可以经过重复一条边或一个点多次的。

现在给出每一次询问，你要按顺序回答 $Yes$ 或者 $No$ ，代表询问的答案。

解题思路

---

先考虑不加边 $(x,y)$ 时应该怎么做。

我们是可以重复经过一条边的，也就是说我们是可以在两个点之间重复行走来获得步数。

还注意到这是一颗树，在没有加边的条件下，我们只有通过走 $a\to b$ 这一条最短路径来到达点 $b$ ，当步数不足时再考虑凑步数。

假设这一条最短路径长度为 $x$，那么显然，我们可以在 $a\to b$ 的途中往返相邻的两个点一次，使得路径长度变成 $x+2$，同理，我们也可以变成 $x+4,x+6,...$ 。

那么我们只需要对最短路径长度 $x$ 判断一下即可：

• 凑步数的前提是 $x\le k$ 。
• 往返时候 $x$ 的奇偶性是不会改变的，即 $x$ 要与 $k$ 同奇偶时我们才可能通过往返的方式凑出 $k$ 步。
• 否则我们就不能恰好 $k$ 步到达 $b$ 。

那么想到了这一步，这题就好做了，判个奇偶和两点路径与 $k$ 的大小即可。

考虑加上边 $(x,y)$ 之后要怎么做。

如果我们 $a\to b$ 的最短路径长度 $x$ 与 $k$ 奇偶性不同时，按照原路径我们显然不能凑出 $k$ 步。

我们考虑经过 $(x,y)$ 这条边能不能使得 $x$ 的奇偶性改变，变成 $x+1,x+3,x+\mathrm{5...}$ 其中之一。

那么我们经过 $(x,y)$ 这条边显然有两种方法：

• 一种是 $a\to x\to y\to b$ 。
• 一种是 $a\to y\to x\to b$ 。

那么同理，我们只需要求出这两种方法的最短路径长度 $x^{\prime }$ 然后判断 $x^{\prime }$ 的大小以及 $x^{\prime }$ 和 $k$ 的奇偶性是否相同即可，对于长度不足的部分，可以通过往返来补足。

求树上两点之间的距离，这里我们只需要用 $LCA$ 就好了。

这里我使用的是树剖求 $LCA$ ，比倍增的空间和时间优秀一些。

时间复杂度：$O(q\log n)$

AC代码：

---

#include <bits/stdc++.h>
#define YES return void(cout << "Yes\n")
#define NO return void(cout << "No\n")
using namespace std;

using u64 = unsigned long long;
using PII = pair<int, int>;
using i64 = long long;

const int N = 1e5 + 10;

vector<int> g[N];

int fa[N], siz[N], son[N], top[N], dep[N], idx;

void DFS1(int u, int ufa) {
    siz[u] = 1, dep[u] = dep[ufa] + 1, fa[u] = ufa;
    for (auto& v : g[u]) {
        if (v == ufa) continue;
        DFS1(v, u);
        siz[u] += siz[v];
        if (siz[v] > siz[son[u]]) son[u] = v;
    }
}

void DFS2(int u, int tp) {
    top[u] = tp;
    if (son[u]) DFS2(son[u], tp);
    for (auto& v : g[u]) {
        if (v == son[u] || v == fa[u]) continue;
        DFS2(v, v);
    }
}

int LCA(int x, int y) {
    while (top[x] != top[y]) {
        if (dep[top[x]] < dep[top[y]]) swap(x, y);
        x = fa[top[x]];
    }
    if (dep[x] > dep[y]) swap(x, y);
    return x;
}

int dist(int x, int y) {
    return dep[x] + dep[y] - 2 * dep[LCA(x, y)];
}

void solve() {

    int n; cin >> n;
    for (int i = 1; i <= n - 1; ++i) {
        int u, v; cin >> u >> v;
        g[u].emplace_back(v);
        g[v].emplace_back(u);
    }

    //树剖求LCA
    DFS1(1, 0);
    DFS2(1, 1);

    int k;
    auto check = [&](int len) {
        return len <= k && k % 2 == len % 2;
    };

    int q; cin >> q;
    for (int i = 1; i <= q; ++i) {
        int x, y, a, b;
        cin >> x >> y >> a >> b >> k;
        int ax = dist(x, a), xb = dist(x, b);
        int ay = dist(y, a), yb = dist(y, b);
        int ab = dist(a, b);
        //三种情况 (a,b) / (a, x, y, b) / (a, y, x, b) 满足一种即可
        if (check(ab) || check(ax + yb + 1) || check(ay + xb + 1)) cout << "Yes\n";
        else cout << "No\n";
    }
}

signed main() {

    ios::sync_with_stdio(0);
    cin.tie(0), cout.tie(0);

    int t = 1; //cin >> t;
    while (t--) solve();

    return 0;
}