1-Trees and Queries (最近公共祖先)

1-Trees and Queries

题目链接

大致题意:

给一个包含n个点的无向图,询问m次,每次把x,y两点连接,问a到b是否存在一条长度为k的路径

每次询问中x,y连边对其他询问不产生影响

---

解题思路:

我们需要用到a点到b点的最短距离,用lca实现,然后就是分情况讨论:

1.不需要x,y这条边:

------①如果a点到b点的最短距离>k的话,NO

------②如果a点到b点的最短距离=k的话,YES

------③如果a点到b点的最短距离<k的话,再分情况:我们可以一条边走多次,也就是说当我们从a
走到b,还剩k-dis(a,b),此时我们随便走一条与路-径上点相连的边,一去一回是走了两次边,也就
是说如果(k-dis(a,b))%2==0,YES,否则NO

2.需要x,y这条边

------①如果点到b点的最短距离<k的话,我们就要用到x,y这条边,此时距离应该是
dis(a,x)+dis(y,b)+1或者dis(a,y)+dis(x,b)+1 ,如果dis(a,x)+dis(y,b)+1或者dis(a,y)+dis(x,b)+1>k,NO

------②如果dis(a,x)+dis(y,b)+1或者dis(a,y)+dis(x,b)+1=k,YES

------③如果dis(a,x)+dis(y,b)+1或者dis(a,y)+dis(x,b)+1<k,再分情况:此时与上面同理,判断(k-
dis(a,x)+dis(y,b)+1或者dis(a,y)+dis(x,b)+1)%2==0,YES,否则NO

简单来说,就是如果k>=a到b距离&&(k-a到b距离)%2==0 YES ,否则NO

---

AC代码:

#include <bits/stdc++.h>
#define rep(i, n) for (int i = 1; i <= (n); ++i)
#define debug(a) cout << #a << " = " << a << endl;
using namespace std;
typedef long long ll;
const int N = 1e5 + 10;
int n, m;
int depth[N], f[N][25];
vector<int>e[N];
void dfs(int u, int fa) {
	depth[u] = depth[fa] + 1;
	f[u][0] = fa;
	for (int i = 1; i <= 24; ++i)f[u][i] = f[f[u][i - 1]][i - 1];
	for (auto& v : e[u]) {
		if (v == fa)continue;
		dfs(v, u);
	}
}
int lca(int a, int b) {
	if (depth[a] < depth[b])swap(a, b);
	for (int k = 24; k >= 0; --k)
		if (depth[f[a][k]] >= depth[b])
			a = f[a][k];
	if (a == b)return a;
	for (int k = 24; k >= 0; --k)
		if (f[a][k] != f[b][k]) {
			a = f[a][k]; b = f[b][k];
		}
	return f[a][0];
}
int dis(int a, int b) {
	int p = lca(a, b);
	return depth[a] + depth[b] - 2 * depth[p];
}
int main(void)
{
	cin >> n;
	for (int i = 1; i < n; ++i) {
		int a, b; cin >> a >> b;
		e[a].push_back(b); e[b].push_back(a);
	}
	dfs(1, 0);
	cin >> m;  //查询
	while (m--) {
		int x, y, a, b, k; cin >> x >> y >> a >> b >> k;
		int d1 = dis(a, b), d2 = dis(a, x) + dis(y, b) + 1, d3 = dis(a, y) + dis(x, b) + 1;
		if (k >= d1 && (k - d1) % 2 == 0)puts("YES");
		else if (k >= d2 && (k - d2) % 2 == 0)puts("YES");
		else if (k >= d3 && (k - d3) % 2 == 0)puts("YES");
		else puts("NO");
	}
	return 0;
}

---