【Luogu_P3379】【模板】最近公共祖先（LCA）

链接

模板

#include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>
#include<queue>
#include<stack>
#include<map>
#include<cstring>
using namespace std;

int n, m, s;
int d[500010], dep[500010], f[500010][20];
int head[500100], tot;
struct node{
	int to, next;
}b[5000010];

void add(int x, int y)
{
	b[++tot]=(node){y, head[x]};
	head[x]=tot;
}
void dfs(int x, int fa)
{
	f[x][0]=fa;
	dep[x]=dep[fa]+1;
	for(int i=head[x]; i; i=b[i].next)
	{
		int y=b[i].to;
		if(y==fa)
			continue;
		dfs(y, x);
	}
}
int lca(int x, int y)
{
	if(dep[x]>dep[y])
		swap(x, y);
	int cnt=dep[y]-dep[x], k=19, t=1<<k;
    while(cnt)
    {
    	if(cnt>=t)
    		y=f[y][k], cnt-=t;
    	k--;
    	t/=2;
    }
    if(x==y) return x;
    k=19;
    while(k>=0)
    {
    	if(f[x][k]!=f[y][k])
    	{
    		x=f[x][k];
    		y=f[y][k];
    	}
    	k--;
    }
    return f[x][0];
}
int main(){
	scanf("%d%d%d", &n, &m ,&s);
	for(int i=1; i<n; i++)
	{
		int x, y;
		scanf("%d%d", &x, &y); 
		add(x, y);
		add(y, x);
	}
	dfs(s, 0);
	for(int j=1; j<20; j++)
		for(int i=1; i<=n; i++)
			f[i][j]=f[f[i][j-1]][j-1];
	while(m--)
	{
		int x, y;
		scanf("%d%d", &x, &y);
		printf("%d\n", lca(x, y));
	}
	return 0;
}