仓鼠找sugar

$$\mathrm{仓鼠找sugar}$$

题目链接：$\mathrm{l}\mathrm{u}\mathrm{o}\mathrm{g}\mathrm{u}\mathrm{P}3398$

题目

小仓鼠的和他的基（mei）友（zi） sugar 住在地下洞穴中，每个节点的编号为 $1\sim n$ 。地下洞穴是一个树形结构。这一天小仓鼠打算从从他的卧室（ $a$ ）到餐厅（ $b$ ），而他的基友同时要从他的卧室（ $c$ ）到图书馆（ $d$ ）。他们都会走最短路径。现在小仓鼠希望知道，有没有可能在某个地方，可以碰到他的基友？

小仓鼠那么弱，还要天天被 zzq 大爷虐，请你快来救救他吧！

输入

第一行两个正整数 $n$ 和 $q$ ，表示这棵树节点的个数和询问的个数。

接下来 $n-1$ 行，每行两个正整数 $u$ 和 $v$ ，表示节点 $u$ 到节点 $v$ 之间有一条边。

接下来 $q$ 行，每行四个正整数 $a$ 、 $b$ 、 $c$ 和 $d$ ，表示节点编号，也就是一次询问，其意义如上。

输出

对于每个询问，如果有公共点，输出大写字母“ Y ”；否则输出“ N ”。

样例输入

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

样例输出

Y
N
Y
Y
Y

数据范围

$20\%$ 的数据 $n<=200,q<=200$

$40\%$ 的数据 $n<=2000,q<=2000$

$70\%$ 的数据 $n<=50000,q<=50000$

$100\%$ 的数据 $n<=100000,q<=100000$

提示

本题时限 $1s$ ，内存限制 $128M$ ，因新评测机速度较为接近 NOIP 评测机速度，请注意常数问题带来的影响。

思路

这道题是一道 LCA 。

我们就分别求出两条路的 $LCA$ ，如果一条路的 LCA 在另一条路里面，就说明可以碰到，否则碰不到。

代码

#include<cmath>
#include<cstdio>
#include<algorithm>

using namespace std;

struct node {
	int to, next;
}e[200001];
int n, q, x, y, le[100001], KK, a, b, c, d, fa[21][100001], LCA1, LCA2, dep[100001];

void add(int x, int y) {
	e[++KK] = (node){y, le[x]}; le[x] = KK;
	e[++KK] = (node){x, le[y]}; le[y] = KK;
}

void dfs(int now, int father) {//遍历一次预处理出深度和父亲
	dep[now] = dep[father] + 1;
	fa[0][now] = father;
	for (int i = le[now]; i; i = e[i].next)
		if (e[i].to != father)
			dfs(e[i].to, now);
}

int LCA(int x, int y) {//LCA
	if (dep[x] < dep[y])
		swap(x, y);
	for (int i = log2(n) + 1; i >= 0; i--)
		if (dep[fa[i][x]] >= dep[y])
			x = fa[i][x];
	if (x == y) return x;
	for (int i = log2(n) + 1; i >= 0; i--)
		if (fa[i][x] != fa[i][y]) {
			x = fa[i][x];
			y = fa[i][y];
		}
	return fa[0][x];
}

int dis(int x, int y) {
	int LCA3 = LCA(x, y);
	return dep[x] - dep[LCA3] + dep[y] - dep[LCA3];
}

int main() {
	scanf("%d %d", &n, &q);//读入
	for (int i = 1; i < n; i++) {
		scanf("%d %d", &x, &y);
		add(x, y);//建图
	}
	
	dfs(1, 0);//预处理
	
	for (int i = 1; i <= 20; i++)//继续预处理，求出倍增用的父亲
		for (int j = 0; j <= n; j++)
			fa[i][j] = fa[i - 1][fa[i - 1][j]];
	
	for (int i = 1; i <= q; i++) {
		scanf("%d %d %d %d", &a, &b, &c, &d);//读入
		LCA1 = LCA(a, b);//求出两条路的LCA
		LCA2 = LCA(c, d);
		if ((dis(a, LCA2) + dis(b, LCA2) == dis(a, b)) || (dis(c, LCA1) + dis(d, LCA1) == dis(c, d))) printf("Y\n");//有一个路径的LCA在另一个路径上
			else printf("N\n");
	}
	
	return 0;
}