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