CF519E A and B and Lecture Rooms

题目链接

题目

见链接。

题解

知识点：倍增，LCA，树型dp。

要找到距离两点 \(u,v\) 相同的点个数，我可以分类讨论：

1. \(u,v\) 是同一个点，那么全部点都可以。
2. \(u,v\) 处于相同深度，那么就是全部点减去 \(LCA(u,v)\) 的 \(u,v\) 两点所在子树的全部点。
3. \(u,v\) 不在相同深度，当 \(u,v\) 距离为奇数时无解，否则解为 \(u,v\) 路径中点为根的子树全部点减去中点的 \(u,v\) 中是中点子孙的点所在子树的全部点。

其中，查询过程用倍增很容易实现，子树大小用树型dp即可。

时间复杂度 \(O((n+m) \log n)\)

空间复杂度 \(O(n \log n)\)

代码

#include <bits/stdc++.h>
using namespace std;
using ll = long long;

struct Graph {
    struct edge {
        int v, nxt;
    };
    int idx;
    vector<int> h;
    vector<edge> e;

    Graph(int n = 0, int m = 0) { init(n, m); }

    void init(int n, int m) {
        idx = 0;
        h.assign(n + 1, 0);
        e.assign(m + 1, {});
    }

    void add(int u, int v) {
        e[++idx] = { v,h[u] };
        h[u] = idx;
    }
};

const int N = 100007;
Graph g;

int dep[N], f[27][N], sz[N];
void dfs(int u, int fa) {
    dep[u] = dep[fa] + 1;
    f[0][u] = fa;
    sz[u] = 1;
    for (int i = 1;i <= 18;i++)
        f[i][u] = f[i - 1][f[i - 1][u]];
    for (int i = g.h[u];i;i = g.e[i].nxt) {
        int v = g.e[i].v;
        if (v == fa) continue;
        dfs(v, u);
        sz[u] += sz[v];
    }
}

int LCA(int u, int v) {
    if (dep[u] < dep[v]) swap(u, v);
    for (int i = 20;i >= 0;i--) {
        if (dep[f[i][u]] >= dep[v])u = f[i][u];
        if (u == v) return u;
    }
    for (int i = 20;i >= 0;i--) {
        if (f[i][u] != f[i][v]) {
            u = f[i][u];
            v = f[i][v];
        }
    }
    return f[0][u];
}

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

int get_ans(int u, int v) {
    if (u == v) return sz[1];
    if (dep[u] == dep[v]) {
        for (int i = 20;i >= 0;i--) {
            if (f[i][u] != f[i][v]) {
                u = f[i][u];
                v = f[i][v];
            }
        }
        return sz[1] - sz[u] - sz[v];
    }
    int dis = dist(u, v);
    if (dis & 1) return 0;
    if (dep[u] < dep[v]) swap(u, v);
    int d = dep[u] - dis / 2;
    for (int i = 20;i >= 0;i--)
        if (dep[f[i][u]] > d) u = f[i][u];
    return sz[f[0][u]] - sz[u];
}

int main() {
    std::ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
    int n;
    cin >> n;
    g.init(n, n << 1);
    for (int i = 1;i <= n - 1;i++) {
        int u, v;
        cin >> u >> v;
        g.add(u, v);
        g.add(v, u);
    }

    dfs(1, 0);

    int m;
    cin >> m;
    while (m--) {
        int u, v;
        cin >> u >> v;
        cout << get_ans(u, v) << '\n';
    }
    return 0;
}