题目描述
s i y i s s s siyisss siyisss 的王国是由 n n n 个村镇和 n − 1 n−1 n−1 条双向道路构成的,村镇从 1 1 1 到 n n n 依次编号,每条双向道路连接两个不同的村镇,使得从任意一个村镇出发都可以到达任意一个村镇。接下来请你回答 q q q 个问题,每次给出两个整数 c c c, d d d,表示现在分别有一个人在村镇 c c c,一个人在村镇 d d d,现在在 c c c 的人要用最短的时间到达村镇 d d d,在村镇 d d d 的人要以最短的时间到达村镇$ c$,假设两人同时出发,两人的速度也是一样的,每条双向道路的长度也是一样的,请问两人相遇的时候是在某一个村镇,还是在某条双向道路上?
输入描述
第一行输入两个整数 n n n, q q q 代表村镇的数量和询问的数量
接下来 n − 1 n−1 n−1 行,每行两个整数用来描述一条双向道路
最后 q q q,每行两个整数代表 c c c, d d 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 ≤ n ≤ 100000 2≤n≤100000 2≤n≤100000
1 ≤ q ≤ 100000 1≤q≤100000 1≤q≤100000
对于每一个询问 1 ≤ c i < d i ≤ n 1≤c_i<d_i≤n 1≤ci<di≤n
解题思路
题意是给出一张无向连通图,然后询问 q q q次,每次要求判断 c c c, d d d两人相遇的位置。
根据数据范围,我们需要将每次询问的时间复杂度降至 O ( 1 ) O(1) O(1)或者 O ( l o g q ) O(log\ q) O(log q)。
理解题意之后,第一个问题就是如何判断两人相遇的位置:
(1)如果最短路的长度为奇数,那么两人在Road
相遇;
(2)如果最短路的长度为偶数,那么两人在Town
相遇。
关于最短路的算法,我们最容易想到的是Floyd。但是 n ∈ [ 2 , 100000 ] n\in [2,100000] n∈[2,100000],直接 T T T飞掉。
回顾题意: n n n个节点、 n − 1 n-1 n−1条边、全连通。
这不就是一棵树嘛。
那么我们要知道树的特点:任意两个节点之间只有一条通路。
我们只需要找出最近公共父节点(LCA),就可以得出两个节点之间的通路长度。
采用倍增寻访算法,时间复杂度为 O ( q l o g q ) O(qlog\ q) O(qlog q),可以接受。
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;
}