[Daimayuan] Collision（C++，多源最短路）

题目描述

$siyisss$ 的王国是由 $n$ 个村镇和 $n-1$ 条双向道路构成的，村镇从 $1$ 到 $n$ 依次编号，每条双向道路连接两个不同的村镇，使得从任意一个村镇出发都可以到达任意一个村镇。接下来请你回答 $q$ 个问题，每次给出两个整数 $c$, $d$，表示现在分别有一个人在村镇 $c$,一个人在村镇 $d$，现在在 $c$ 的人要用最短的时间到达村镇$d$，在村镇 $d$ 的人要以最短的时间到达村镇$ c$，假设两人同时出发，两人的速度也是一样的，每条双向道路的长度也是一样的，请问两人相遇的时候是在某一个村镇，还是在某条双向道路上？

输入描述

第一行输入两个整数 $n$, $q$ 代表村镇的数量和询问的数量

接下来 $n-1$ 行，每行两个整数用来描述一条双向道路

最后 $q$，每行两个整数代表 $c$, $d$

输出描述

对于每个询问，如果他们在某个村镇相遇，请示出Town，否则输出Road

样例输入1

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

样例输出1

Town
Town

样例输入2

9 9
2 3
5 6
4 8
8 9
4 5
3 4
1 9
3 7
7 9
2 5
2 6
4 6
2 4
5 8
7 8
3 6
5 6

样例输处2

Town
Road
Town
Town
Town
Town
Road
Road
Road

数据范围

$2\le n\le 100000$

$1\le q\le 100000$

对于每一个询问 $1\le c_i<d_i\le n$

解题思路

题意是给出一张无向连通图，然后询问$q$次，每次要求判断$c$,$d$两人相遇的位置。

根据数据范围，我们需要将每次询问的时间复杂度降至$O(1)$或者$O(logq)$。

理解题意之后，第一个问题就是如何判断两人相遇的位置：

（1）如果最短路的长度为奇数，那么两人在Road相遇；

（2）如果最短路的长度为偶数，那么两人在Town相遇。

关于最短路的算法，我们最容易想到的是Floyd。但是$n\in [2,100000]$，直接$T$飞掉。

回顾题意：$n$个节点、$n-1$条边、全连通。

这不就是一棵树嘛。

那么我们要知道树的特点：任意两个节点之间只有一条通路。

我们只需要找出最近公共父节点（LCA），就可以得出两个节点之间的通路长度。

采用倍增寻访算法，时间复杂度为$O(qlogq)$，可以接受。

AC代码如下：

#include <iostream>
#include <cstdlib>
#include <cstdio>
#include <string.h>
using namespace std;
const int max_n = 1e5;
const int max_expo = 32;

struct edge { int v, next; }edges[max_n * 2];
int tot = -1, heads[max_n + 1];
int dp[max_n + 1][max_expo];
int lg[max_n + 1];
int depth[max_n + 1];


void add_edge(int u, int v) {
	edges[++tot] = { v,heads[u] }; heads[u] = tot;
	edges[++tot] = { u,heads[v] }; heads[v] = tot;
}

void init(int s, int fa) {
	for (int i = 1; i < lg[depth[s]]; i++) {
		dp[s][i] = dp[dp[s][i - 1]][i - 1];
	}

	for (int i = heads[s]; i != -1; i = edges[i].next) {
		int v = edges[i].v;
		if (v != fa) {
			dp[v][0] = s;
			depth[v] = depth[s] + 1;
			init(v, s);
		}
	}
}

int lca(int x, int y) {
	if (depth[x] < depth[y]) swap(x, y);
	while (depth[x] != depth[y]) x = dp[x][lg[depth[x] - depth[y]] - 1];
	if (x == y) return x;
	for (int i = lg[depth[x]]; i >= 0; i--) {
		if (dp[x][i] != dp[y][i]) {
			x = dp[x][i];
			y = dp[y][i];
		}
	}
	return dp[x][0];
}

int main() {
	int n, q, u, v;
	//cin >> n >> q;
	scanf("%d%d", &n, &q);
	for (int i = 1; i <= max_n; i++) {
		lg[i] = lg[i - 1] + (1 << lg[i - 1] == i);
	}
	memset(heads + 1, -1, sizeof(int) * n);
	for (int i = 0; i < n - 1; i++) {
		//cin >> u >> v;
		scanf("%d%d", &u, &v);
		add_edge(u, v);
	}

	dp[1][0] = 1;
	depth[1] = 1;
	init(1, 1);

	for (int i = 0; i < q; i++) {
		//cin >> u >> v;
		scanf("%d%d", &u, &v);
		int ret = lca(u, v);
		//if ((depth[ret] * 2 - depth[u] - depth[v]) % 2 == 0) cout << "Town" << endl;
		//else cout << "Road" << endl;
		if ((depth[ret] * 2 - depth[u] - depth[v]) % 2 == 0) printf("Town\n");
		else printf("Road\n");
	}
	return 0;
}