最近公共祖先(LCA)---倍增法

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<cstring>
#include<algorithm>
#define Size 1000000
using namespace std;
int be[Size],ne[Size],to[Size],e;
void add(int x,int y){to[++e]=y;ne[e]=be[x];be[x]=e;}
int deep[Size]; 
int fa[Size>>1][20];
bool vis[Size];
int read(){
	int x=0;
	char ch=getchar();
	while(ch<'0' || ch>'9')ch=getchar();
	while(ch>='0' && ch<='9'){
		x=x*10+ch-'0';
		ch=getchar();
	}
	return x;
}
void dfs(int x){
	vis[x]=1;
	for(int i=be[x];i;i=ne[i]){
		if(!vis[to[i]]){
			fa[to[i]][0]=x;//标记上一个为父亲节点 
			deep[to[i]]=deep[x]+1; 
			dfs(to[i]);
		}
	}
}
void build(int n){
	int k=log(n)/log(2);
	for(int i=1;i<=k;i++)
		for(int j=1;j<=n;j++)
			if(fa[j][i-1]!=0 && fa[fa[j][i-1]][i-1]!=0)
				fa[j][i]=fa[fa[j][i-1]][i-1]; 
}
int query(int x,int y){
	if(deep[x]<deep[y])swap(x,y);//把x作为深度较大的点 
	int t;
	for(t=0;(1<<t)<=deep[x];t++);
	t--;
	for(int j=t;j>=0;j--)
		if(deep[x]-(1<<j)>=deep[y])
			x=fa[x][j];
	if(x==y)return x;
	for(int j=t;j>=0;j--){
		if(fa[x][j]!=0 && fa[x][j]!=fa[y][j]){//不能为同一个点 
			x=fa[x][j];
			y=fa[y][j];
		}		
	}
	return fa[x][0];
}
int main(){
	int n,m,s;//n个点,m个边,以s点为根节点 
	n=read();m=read();s=read();
	for(int i=1;i<n;i++){
		int x,y;
		x=read();y=read();
		add(x,y);add(y,x);
	}
	deep[s]=1;
	dfs(s);
	build(n);
	for(int i=1;i<=m;i++){
		int x,y;
		x=read();y=read();
		printf("%d\n",query(x,y));
	}
	return 0;
}
/*
5 5 4
3 1
2 4
5 1
1 4
2 4
3 2
3 5
1 2
4 5

4
4
1
4
4
*/

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值