洛谷 P4281 [AHOI2008]紧急集合 / 聚会（树上倍增 LCA）

[AHOI2008]紧急集合 / 聚会

题目描述

欢乐岛上有个非常好玩的游戏，叫做“紧急集合”。在岛上分散有 $n$ 个等待点，有 $n-1$ 条道路连接着它们，每一条道路都连接某两个等待点，且通过这些道路可以走遍所有的等待点，通过道路从一个点到另一个点要花费一个游戏币。

参加游戏的人三人一组，开始的时候，所有人员均任意分散在各个等待点上（每个点同时允许多个人等待），每个人均带有足够多的游戏币（用于支付使用道路的花费）、地图（标明等待点之间道路连接的情况）以及对话机（用于和同组的成员联系）。当集合号吹响后，每组成员之间迅速联系，了解到自己组所有成员所在的等待点后，迅速在 $n$ 个等待点中确定一个集结点，组内所有成员将在该集合点集合，集合所用花费最少的组将是游戏的赢家。

小可可和他的朋友邀请你一起参加这个游戏，由你来选择集合点，聪明的你能够完成这个任务，帮助小可可赢得游戏吗？

输入格式

第一行两个正整数 $n$ 和 $m$，分别表示等待点的个数（等待点也从 $1$ 到 $n$ 进行编号）和获奖所需要完成集合的次数。

随后 $n-1$ 行，每行两个正整数 $a,b$，表示编号为 $a$ 和编号为 $b$ 的等待点之间有一条路。

随后 $m$ 行，每行用三个正整数 $x,y,z$，表示某次集合前小可可、小可可的朋友以及你所在等待点的编号。

输出格式

输出共 $m$ 行，每行两个用空格隔开的整数 $p,c$。其中第 $i$ 行表示第 $i$ 次集合点选择在编号为 $p$ 的等待点，集合总共的花费是 $c$ 个游戏币。

样例 #1

样例输入 #1

6 4  
1 2  
2 3  
2 4 
4 5
5 6
4 5 6
6 3 1
2 4 4 
6 6 6

样例输出 #1

5 2
2 5
4 1
6 0

提示

对于 $40\%$ 的数据，$n\le 2\times 10^3$，$m\le 2\times 10^3$。

对于 $100\%$ 的数据，$1\le x,y,z\le n\le 5\times 10^5$，$1\le m\le 5\times 10^5$。

题意：

给定 n 个点，共 n - 1 条边的有根树，以及 m 次询问，每次询问给出 3 个点，要求在树中 找到一点，使得 3 个点到达该点的距离之和最小，输出 该点编号以及距离之和。

思路：

如果不是标签，我可能还想不到是用 LCA 来做。

接下来直接上思路。

我们在纸上模拟一遍样例，我们可以发现：

• 当 a、b、c 三个点中有两个点的 lca 是同一个点时（重合），那么我们选择的点则为另一个点与这两个点中任意一点的 lca，举个例子，假如 a、b 的 lca 均为 d，那么我们要选择的点即为 c、d 的 lca。
• 当 a、b、c 三个点的 lca 各不相同 或者 都相同，那么 我们选择的点 即为任意两点的 lca。

我在纸上模拟并发现这一点还挺不容易的，可以将这一发现当做一个结论记住。

剩下的步骤，就是对应情况算出距离并输出即可。

代码：

套 树上倍增 lca 板子就可以了，注意本题显然 在线求 lca 更加方便，貌似不能用 tarjan 求 lca

一开始总是 TLE 在 8、9，原因在于：没必要开一个 长度为 5e5 的 cnt 数组 来表示 3 个点的 lca 重合情况，导致每次查询都要将这么大的数组 以 O(n) 的时间清空一遍，显然会 T。（积累经验，下次不再犯）

#define _CRT_SECURE_NO_WARNINGS 1
#include <bits/stdc++.h>

using namespace std;
//#define int long long
//#define map unordered_map
const int N = 5e5 + 10, M = N << 1;
int n, m;
int h[N], e[M], ne[M], idx;
int depth[N], fa[N][20];
int dist[N];

void add(int a, int b)
{
	e[idx] = b, ne[idx] = h[a], h[a] = idx++;
}

void dfs(int u, int father)
{
	for (int i = h[u]; ~i; i = ne[i])
	{
		int j = e[i];
		if (j == father) continue;
		dist[j] = dist[u] + 1;
		dfs(j, u);
	}
}

void bfs(int root)
{
	queue<int> q; q.push(root);
	while (q.size())
	{
		auto t = q.front(); q.pop();
		for (int i = h[t]; ~i; i = ne[i])
		{
			int j = e[i];
			if (depth[j] > depth[t] + 1)
			{
				depth[j] = depth[t] + 1;
				fa[j][0] = t;
				for (int k = 1; k <= 19; ++k)
				{
					fa[j][k] = fa[fa[j][k - 1]][k - 1];
				}
				q.push(j);
			}
		}
	}
}

void init(int root)
{
	memset(depth, 0x3f, sizeof depth);
	depth[0] = 0, depth[root] = 1;
	bfs(root);
}

int lca(int x, int y)
{
	if (depth[x] < depth[y])
	{
		swap(x, y);
	}
	for (int k = 19; k >= 0; --k)
	{
		if (depth[fa[x][k]] >= depth[y]) x = fa[x][k];
	}

	if (x == y) return x;
	for (int k = 19; k >= 0; --k)
	{
		if (fa[x][k] != fa[y][k])
		{
			x = fa[x][k];
			y = fa[y][k];
		}
	}
	return fa[x][0];
}

int getdis(int a, int b)
{
	return dist[a] + dist[b] - 2 * dist[lca(a, b)];
}

signed main()
{
	int T = 1; //cin >> T;

	while (T--)
	{
		scanf("%d%d", &n, &m);
		memset(h, -1, sizeof h);
		int t = n - 1, root = 0;
		for (int i = 0; i < t; ++i)
		{
			int a, b; scanf("%d%d", &a, &b);
			if (!i) root = a;
			add(a, b), add(b, a);
		}
		init(root);
		dfs(root, -1);
		while (m--)
		{
			int x, y, z; scanf("%d%d%d", &x, &y, &z);

			int p1 = lca(x, y);
			int p2 = lca(y, z);
			int p3 = lca(x, z);
			int res = 0;
			int point = 0;
			
			if (p1 == p2 && p1 != p3)
			{
				int d1 = getdis(x, p3);
				int d2 = getdis(y, p3);
				int d3 = getdis(z, p3);
				res = d1 + d2 + d3;
				point = p3;
			}
			else if (p1 == p3 && p1 != p2)
			{
				int d1 = getdis(x, p2);
				int d2 = getdis(y, p2);
				int d3 = getdis(z, p2);
				res = d1 + d2 + d3;
				point = p2;
			}
			else if(p2 == p3 && p2 != p1)
			{
				int d1 = getdis(x, p1);
				int d2 = getdis(y, p1);
				int d3 = getdis(z, p1);
				res = d1 + d2 + d3;
				point = p1;
			}
			else
			{
				res = getdis(x, p1) + getdis(y, p1) + getdis(z, p1);
				point = p1;
			}
			printf("%d %d\n", point, res);
		}
	}

	return 0;
}