平日里最喜欢做的事就是蒸发学水。
【题目描述】
小 X 所在的城市 X 城是一个含有 N 个节点的无向图,同时,由于 X 国是一
个发展中国家,为了节约城市建设的经费,X 国首相在建造 X 城时只建造 N – 1
条边,使得城市的各个地点能够相互到达。
小 X 计划蒸发 Q 天的学水,每一天会有一名学水从 A 地走到 B 地,并在沿
途各个地点留下一个水塘。此后,小 X 会从 C 地走到 B 地,并用佛光蒸发沿途
的水塘。由于 X 城是一个学佛横行的城市,学水留下的水塘即使没有被小 X 蒸
发,也会在第二天之前被其他学佛蒸发殆尽。
现在,小 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 能够蒸发的水塘数。
【样例 1 输入】
3 3 1
1 2
2 3
1 2 3
1 1 3
3 1 3
【样例 1 输出】
1
1
3
分析:这道题的本质就是要求树上同一个点为起点到两个端点的路径交,暴力的复杂度是O(nq)的,可以过50%的点,因为涉及到路径,如果用树链剖分可以过16个点,不过不能用memset,要用时间戳。正解非常巧妙,其实对于树上的任意三点只有可能是两种情况:一条链或者以某个点为中心.
显然对于第二种情况,我们只需要求出中心到B点的距离就好了,如果是一条链我们该怎么做呢?这个时候我们可以把中心定义为深度为中位数的那个点,结合两个中心的定义,我们可以脑补出怎么求中心:ABC三个点两两求LCA,其中深度最大的LCA就是中心,最后求一下中心到B点的距离就好了。
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <queue> using namespace std; const int maxn = ; int n, q, num,head[maxn],to[maxn * ],nextt[maxn * ],tot = ,dist[maxn],fa[maxn][]; 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][] = u;
dfs(v, u, d + );
}
}
} int lca(int x, int y)
{
if (x == y)
return x;
if (dist[x] < dist[y])
swap(x, y);
for (int i = ;i >= ; i--)
if (dist[fa[x][i]] >= dist[y])
x = fa[x][i];
if (x == y)
return x;
for (int i = ; i >= ; i--)
if (fa[x][i] != fa[y][i])
{
x = fa[x][i];
y = fa[y][i];
}
return fa[x][];
} int main()
{
scanf("%d%d%d", &n, &q, &num);
for (int i = ; i < n; i++)
{
int x, y;
scanf("%d%d", &x, &y);
add(x, y);
add(y, x);
}
dfs(, , );
for (int j = ; j <= ; j++)
for (int i = ; i <= n;i++)
fa[i][j] = fa[fa[i][j - ]][j - ];
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] - * dist[lca(d, b)] + );
} return ;
}