样例能过但全 WA 求调
查看原帖
样例能过但全 WA 求调
1088732
GXZJQ楼主2024/6/12 18:51

感谢大佬 Orz

/**
 * 题目名称:P8805 [蓝桥杯 2022 国 B] 机房
 * 题目链接:https://www.luogu.com.cn/problem/P8805
 * 做题时间:2024年6月8日
 * 考查内容:最近公共祖先
 * 难度评级:普及+/提高
 * 主要思路:注意建图的方式即可
 * 注意事项:
 **/
#include<bits/stdc++.h>
using namespace std;
const int N = 1e6 + 10;
int d[N], sum[N][31];
int anc[N][31];
int h[N], to[N], nxt[N], cnt;
int n, m, x, y;
int a[N];

void add(int u, int v) {
	to[++cnt] = v;
	nxt[cnt] = h[u];
	h[u] = cnt;
	a[u]++;
}

void DFS(int u, int fa) {
	d[u] = d[fa] + 1;
	anc[u][0] = fa;
	sum[u][0] = a[u];
	for(int j = 1; j < 31; j++) {
		anc[u][j] = anc[anc[u][j - 1]][j - 1];
		sum[u][j] = sum[anc[u][j - 1]][j - 1] + sum[u][j - 1];
	}
	for (int i = h[u]; i != -1; i = nxt[i]) {//遍历这个节点的所有出边
		int v = to[i];//储存一下边 u -> v 的终点
		if (v != fa) DFS(v, u);//搜索这个点的子节点
	}
}

int LCA(int u, int v) {
	int ans = 0;
	if (d[u] < d[v]) swap(u, v);
	for (int i = 30; i >= 0; i--) if (d[anc[u][i]] >= d[v]) ans += sum[u][i], u = anc[u][i];//先将较深的节点提高至一个高度
	if (u == v) return ans + a[u];
	for (int i = 30; i >= 0; i--) if (anc[u][i] != anc[v][i]) u = anc[u][i], v = anc[v][i], ans += sum[u][i] + sum[v][i];//将二者提升到最近公共祖先的下一层
	ans += sum[u][0] + sum[v][0] + sum[anc[u][0]][0];
	return ans;
}

int main() {
//	freopen(".in", "r", stdin);
//	freopen(".out", "w", stdout);
	ios::sync_with_stdio(false);
	cin.tie(0), cout.tie(0);
	memset(h, -1, sizeof h);
	cin >> n >> m;
	for (int i = 1; i <= n - 1; i++) {
		int u, v;
		cin >> u >> v;
		add(u, v), add(v, u);
	}
	DFS(1, 0);
//	cout << "延迟分别为:"; 
//	for (int i = 1; i <= n; i++) cout << a[i] << " ";
//	cout << endl;
	for (int i = 1; i <= m; i++) {
		int u, v;
		cin >> u >> v;
		cout << LCA(u, v) << endl;
	}
	return 0;
}
2024/6/12 18:51
加载中...