「2024 - 暑假 - Day-4 提高笔记-LCA最近公共祖先」

合集 - C++信息竞赛(17)

1.CSP考前2023-10-042.「Day 1—递归问题」2024-08-043.「Day 2—贪心问题&分治&前缀和」2024-08-054.「Day 3—深度优先搜索 & 广度优先搜索」2024-08-065.「Day 4—图的存储 & 图上搜索」2024-08-116.「Day 5—最短路径」2024-08-117.「Day 6—单调栈 & 单调队列 & 并查集」2024-08-118.「Day 7—离散化 & 树状数组 & 线段树」2024-08-129.「Day 8—最小生成树之Kruskal & Prim」2024-08-1210.「Day 9 & 10—DP问题」2024-08-1411.「Day 11 & 12 & 13 & 14—杂项」2024-08-1712.「2024 - 暑假 - Day-1 提高笔记-割点(割边)」2024-10-0913.「2024 - 暑假 - Day-2 提高笔记-字典树」2024-10-1614.【基础知识】2024-10-1815.「2024 - 暑假 - Day-3 提高笔记-ST表 & RMQ」2024-10-22

16.「2024 - 暑假 - Day-4 提高笔记-LCA最近公共祖先」2024-10-22

17.「2025 - 寒假 - Day-2 提高笔记-反悔贪心」01-22

收起

#include<iostream>
using namespace std;

const int MAXN = 5 * 1e5 + 5;
struct node{
	int to,next;
}e[MAXN * 2];
int f[MAXN][20],dp[MAXN];//f[x][i] 表示 x 的第2^i个节点的编号 
int h[MAXN * 2],tot = 0;
int n,m,s;
void add(int x,int y){
	e[++ tot] = {y,h[x]};
	h[x] = tot;
}
void dfs(int x,int fa){
	dp[x] = dp[fa] + 1;
	f[x][0] = fa;//f[x][0] = x的第1(2^0)个父节点的节点编号 
	for(int i = 1;(1 << i) <= dp[x];i ++){
		f[x][i] = f[f[x][i - 1]][i - 1];
	}
	for(int i = h[x];i;i = e[i].next){
		int to = e[i].to;
		if(to != fa){
			dfs(to,x);
		}
	}
	return;
}
int u,v;
int LCA(int x,int y){
	if(dp[x] < dp[y]) swap(x,y);
	for(int i = 19;i >= 0;i --){
		if(dp[x] - (1 << i) >= dp[y]){
			x = f[x][i];
		}
	}
	if(x == y) return x;
	for(int i = 19;i >= 0;i --){
		if(f[x][i] != f[y][i]){
			x = f[x][i];
			y = f[y][i];
		}
	} 
	return f[x][0];
}

int main(){
	cin >> n >> m >> s;
	for(int i = 1;i <= n - 1;i ++){
		cin >> u >> v;
		add(u,v); add(v,u); 
	}
	dfs(s,0);
	for(int i = 1;i <= m;i ++){
		cin >> u >> v;
		cout << LCA(u,v) << '\n';
	}
	return 0;
}