Z0600. [模板]最近公共祖先LCA

Description

给出N，Q .N<=5*10^5

代表一个树有N个点 ,树的根为1

Q代表有Q个询问，询问A，B的最近公共祖先是哪一个.

Format

Input

第一行给出N，Q

下面N-1行描述这个树，格式为A,B，代表A为B的Son

现来下Q行

给出A，B，代表进行询问A，B的LCA是哪一个.

Output

针对每个询问输出结果

Samples

输入数据 1

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

 输出数据 1

2
2

思路

都说了是模板题了，直接套lca的板子就行了

代码

#include<bits/stdc++.h>
#define int long long
using namespace std;
inline int read() {int x=0,f=1;char ch=getchar();while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();return x*f;}
int n,m,k,l,r;
vector<int> a[114514];
int b[114514],ans[114514][22];
void dfs(int x,int fa,int sum){
    b[x] = sum;
    for(int i=0;i<a[x].size();i++){
    	int y = a[x][i];
		if(y==fa) continue;
        ans[y][0] = x;
		dfs(y,x,sum+1);
	}
}
int lca(int x,int y){
	if(b[x]<b[y]) swap(x,y);
	for(int i=20;i>=0;i--) if(b[ans[x][i]]>=b[y]) x = ans[x][i];
	if(x==y) return x;
	for(int i=20;i>=0;i--) if(ans[x][i]!=ans[y][i]) x = ans[x][i],y = ans[y][i];
	return ans[x][0];
}
signed main(){
	n = read();
    m = read();
	for(int i=1;i<n;i++){
		int x,y;
		x = read();
        y = read();
		a[x].push_back(y);
		a[y].push_back(x);
	}
    dfs(1,-1,1);
	for(int i=1;i<=20;i++) for(int j=1;j<=n;j++) ans[j][i] = ans[ans[j][i-1]][i-1];
	for(int i=1;i<=m;i++){
        int x,y;
        x = read();
        y = read();
        printf("%d\n",lca(x,y));
	}
	return 0;
}