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

P3379 【模板】最近公共祖先（LCA）
题目描述
如题，给定一棵有根多叉树，请求出指定两个点直接最近的公共祖先。

输入格式
第一行包含三个正整数 N,M,SN,M,S，分别表示树的结点个数、询问的个数和树根结点的序号。

接下来 N-1N−1 行每行包含两个正整数 x, yx,y，表示 xx 结点和 yy 结点之间有一条直接连接的边（数据保证可以构成树）。

接下来 MM 行每行包含两个正整数 a, ba,b，表示询问 aa 结点和 bb 结点的最近公共祖先。

输出格式
输出包含 MM 行，每行包含一个正整数，依次为每一个询问的结果。

输入输出样例
输入 #1复制
5 5 4
3 1
2 4
5 1
1 4
2 4
3 2
3 5
1 2
4 5
输出 #1复制
4
4
1
4
4
说明/提示
对于 30%30% 的数据，N\leq 10N≤10，M\leq 10M≤10。

对于 70%70% 的数据，N\leq 10000N≤10000，M\leq 10000M≤10000。

对于 100%100% 的数据，N\leq 500000N≤500000，M\leq 500000M≤500000。

样例说明：

该树结构如下：

第一次询问：2, 42,4 的最近公共祖先，故为 44。

第二次询问：3, 23,2 的最近公共祖先，故为 44。

第三次询问：3, 53,5 的最近公共祖先，故为 11。

第四次询问：1, 21,2 的最近公共祖先，故为 44。

第五次询问：4, 54,5 的最近公共祖先，故为 44。

故输出依次为 4, 4, 1, 4, 44,4,1,4,4。

p数组的第二维开小了一个就找了好长时间。。。

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
#define N 500010
struct node{
	int to,next;
}s[N*2];
int head[N],num=0,d[N],p[N][21];
void add(int u,int v){
	s[++num]=(node){v,head[u]};
	head[u]=num;
}
void dfs(int x,int fa){//x是当前节点，fa是父节点 
	d[x]=d[fa]+1;
	p[x][0]=fa;//x前进2^0步是fa 
	for(int i=1;(1<<i)<=d[x];i++){
		p[x][i]=p[p[x][i-1]][i-1];//x前进2^i的步 
	}
	for(int i=head[x];i;i=s[i].next){
		if(fa!=s[i].to) dfs(s[i].to,x);
	}
}  
int lca(int x,int y){
	if(d[x]>d[y]) swap(x,y);
	for(int i=20;i>=0;i--){//y向上爬，使之与x的深度一样 
		if(d[x]<=d[y]-(1<<i))
			y=p[y][i];
	}
	if(x==y) return x;
	for(int i=20;i>=0;i--){//x和y一起上移 
		if(p[x][i]==p[y][i]) continue;
		else{
			x=p[x][i],y=p[y][i];
		}
	}		
	return p[x][0];
}
int main()
{
	int n,m,r,x,y,i,j;
	scanf("%d %d %d",&n,&m,&r);
	for(i=1;i<n;i++){
		scanf("%d %d",&x,&y);
		add(x,y);add(y,x);
	}
	dfs(r,0);
	while(m--){
		scanf("%d %d",&x,&y);
		printf("%d\n",lca(x,y));
	}
	return 0;
}