洛谷 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$。

算法竞赛进阶指南，图论习题10
1、 套用树链剖分计算最近公共祖先LCA
2、 对于三个点 x, y, z ，找到一个点 p ，使得这三点到p点的距离和最小。
	点p必然是某两个点的 LCA。 设 common_lca 是 x, y, z 这三点的公共祖先。
	int a1 = lca(x, y), a2 = lca(y, z), a3 = lca(z, x);
	这里先算出a1 = lca(x, y)， 然后计算 x 到 y 的最短距离就是 
	dep[x] + dep[y] - dep[a1] - dep[a1]。

	然后再计算 a1 和 z 的公共祖先 common_lca = lca(a1, z), z 和 a1 之间的最短距离 = 
	dep[z] - dep[common_lca] + dep[a1] - dep[common_lca]
	
	合并：
	cost = dep[x] + dep[y] - dep[a1] - dep[a1] + dep[z] - dep[common_lca] + dep[a1] - dep[common_lca];
	化简：
	cost = dep[x] + dep[y] + dep[z] - dep[a1] - dep[common_lca] * 2;
	观察：
	如果 dep[a1] 尽可能大， cost 就尽可能小。 因为其他项是定值。
	int a1 = lca(x, y), a2 = lca(y, z), a3 = lca(z, x); 题目就要求 a1, a2, a3 这三个点深度最小的。

#include <bits/stdc++.h>
using namespace std;
const int MaxN = 1e6 + 10;
vector<int> e[MaxN];
int fa[MaxN], dep[MaxN], sz[MaxN]; // sz[u] 存根为u的子树节点数
int son[MaxN]; // son[u] 存u的重儿子
int top[MaxN]; // top[u] 存u所在重链的顶点
int n, m;

// 搞 fa， dep, son
void dfs1(int u, int father)
{
	fa[u] = father, dep[u] = dep[father] + 1, sz[u] = 1;
	for(int v : e[u])
	{
		if(v == father) continue;
		dfs1(v, u);
		sz[u] += sz[v];
		if(sz[son[u]] < sz[v])
			son[u] = v;
	}
}

// 搞 top
void dfs2(int u, int t)
{
	top[u] = t;
	if(!son[u])	// 叶子节点到这里，直接返回 
		return;
	dfs2(son[u], t); // 搜重儿子
	for(int v : e[u])
	{
		if(v == fa[u] || v == son[u])
			continue;
		dfs2(v, v);		// 搜轻儿子，轻儿子都是每条重链的顶点
	}
}

int lca(int u, int v)
{
	while(top[u] != top[v])
	{
		if(dep[top[u]] < dep[top[v]])
			swap(u, v);
		u = fa[top[u]];	// u指向其所在重链顶点 的父节点

	}
	return dep[u] < dep[v] ? u : v;
}

int main()
{
	scanf("%d%d", &n, &m);
	int x, y;
	for(int i = 1; i < n; ++i)
	{
		scanf("%d%d", &x, &y);
		e[x].push_back(y);
		e[y].push_back(x);
	}
	dfs1(1, 0);
	dfs2(1, 1);
	int z;
	int cost;
	while(m--)
	{
		scanf("%d%d%d", &x, &y, &z);
		int a1 = lca(x, y), a2 = lca(y, z), a3 = lca(z, x);
		int common_lca = lca(a1, z);
		int max_lca = a1;
		if(dep[max_lca] < dep[a2])
			max_lca = a2;
		if(dep[max_lca] < dep[a3])
			max_lca = a3;
		cost = dep[x] + dep[y] + dep[z] - dep[max_lca] - dep[common_lca] * 2;
		printf("%d %d\n", max_lca, cost);
	}
	return 0;
}