3415: 【提高】小 X 的佛光

时间限制: 1.000 Sec  内存限制: 128 MB
提交: 14  解决: 7
[命题人:][下载数据: 110]

提交状态报告

题目描述

题目背景】
小 X 是远近闻名的学佛，平日里最喜欢做的事就是蒸发学水。
【题目描述】
小 X 所在的城市 X 城是一个含有 N 个节点的无向图，同时，由于 X 国是一
个发展中国家，为了节约城市建设的经费，X 国首相在建造 X 城时只建造 N – 1
条边，使得城市的各个地点能够相互到达。
小 X 计划蒸发 Q 天的学水，每一天会有一名学水从 A 地走到 B 地，并在沿
途各个地点留下一个水塘。此后，小 X 会从 C 地走到 B 地，并用佛光蒸发沿途
的水塘。由于 X 城是一个学佛横行的城市，学水留下的水塘

输入

输入格式】
从文件 light.in 中读取数据。
第一行三个整数 N、Q、NUM，分别表示 X 城地点的个数，小 X 蒸发学水的
天数，以及测试点编号。注意，测试点编号是为了让选手们更方便的获得部分分，
你可能不需要用到这则信息，在下发的样例中，测试点编号的含义是该样例满足
某一测试点限制。
N – 1 行，每行两个整数 X、Y，表示 X 地与 Y 地之间有一条边。
接下来 Q 行，每行三个整数 A、B、C，表示一天中，有一名学水从 A 地走到
B 地，而小 X 会从 C 地走到 B 地。

输出

输出格式】
输出 Q 行，每行一个整数，表示小 X 能够蒸发的水塘数

样例

输入  复制

输入】 3 3 1 1 2 2 3 1 2 3 1 1 3 3 1 3

输出  复制

提示

【数据范围及子任务】
1：第 i 条边连接第 i 和第 i + 1 个地点。
特殊性质 2：A = C
测试点编号  N  Q  特殊性质 1  特殊性质 2
1
≤10  ≤10
NO  NO
2
3
4
5
≤1000  ≤1000  6
7
8
≤10 5  ≤30  9
10
11
≤10 5  ≤10 5
YES
YES
12
NO
13
14
NO
YES
15
16  NO
17
≤2 * 10 5  ≤2 * 10 5
YES
YES
18  NO
19
NO
YES
20  NO

AC

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <queue>

using namespace std;

const int maxn = 200010;

int n, q, num,head[maxn],to[maxn * 2],nextt[maxn * 2],tot = 1,dist[maxn],fa[maxn][20];

void add(int x, int y)
{
    to[tot] = y;
    nextt[tot] = head[x];
    head[x] = tot++;
}

void dfs(int u, int from, int d)
{
    dist[u] = d;
    for (int i = head[u];i;i = nextt[i])
    {
        int v = to[i];
        if (v != from)
        {
            fa[v][0] = u;
            dfs(v, u, d + 1);
        }
    }
}

int lca(int x, int y)
{
    if (x == y)
        return x;
    if (dist[x] < dist[y])
        swap(x, y);
    for (int i = 19;i >= 0; i--)
        if (dist[fa[x][i]] >= dist[y])
            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 main()
{
    scanf("%d%d%d", &n, &q, &num);
    for (int i = 1; i < n; i++)
    {
        int x, y;
        scanf("%d%d", &x, &y);
        add(x, y);
        add(y, x);
    }
    dfs(1, 0, 1);
    for (int j = 1; j <= 19; j++)
        for (int i = 1; i <= n;i++)
            fa[i][j] = fa[fa[i][j - 1]][j - 1];
    while (q--)
    {
        int a, b, c;
        scanf("%d%d%d", &a, &b, &c);
        int d = lca(a, b), e = lca(b, c), f = lca(a, c);
        if (dist[d] < dist[e])
            swap(d, e);
        if (dist[d] < dist[f])
            swap(d, f);
        printf("%d\n", dist[d] + dist[b] - 2 * dist[lca(d, b)] + 1);
    }

    return 0;
}