洛谷P3379 最近公共祖先(LCA)【树链剖分】

原理及实现:
为了方便描述,称 T x T_x Tx x x x所在树链的顶端节点, D x D_x Dx x x x的深度, D T x D_{T_x} DTx x x x树链顶端节点的深度

两个节点所在树链只有两种情况:
①:在同一条链(即 T x = T y T_x=T_y Tx=Ty),此时 L C A ( x , y ) LCA(x,y) LCA(x,y)为两者深度更浅的。

②:不在同一条链(即 T x ! = T y T_x!=T_y Tx!=Ty),那就让 D T D_{T} DT小的节点跳到所在树链顶端的父节点(因为 T T x = T x T_{T_x}=T_x TTx=Tx),重复此过程直到两者在同一条树链,此时就和情况①一样了
注意,比较是是两个节点树链顶端的深度而不是两个节点自身的深度
在这里插入图片描述
像这种情况, x x x更深,但是 T x T_x Tx更浅,如果跳 x x x的话就跳过头了

#include<iostream>
#include<sstream>
#include<string>
#include<queue>
#include<map>
#include<unordered_map>
#include<set>
#include<vector>
#include<stack>
#include <utility>
#include<algorithm>
#include<cstdio>
#include<list>
#include<cmath>
#include<cstdlib>
#include<cstring>
#include<iomanip>
#include<time.h>
#define int long long
#define PI acos(-1.0)
#define eps 1e-9
#define lowbit(a) ((a)&-(a))
const int mod = 1e9+7;
using namespace std;
inline int read(){
	char c=getchar();int x=0,s=1;
	while(c<'0'||c>'9'){if(c=='-')s=-1;c=getchar();}
	while(c>='0'&&c<='9'){x=x*10+c-'0';c=getchar();}
	return x*s;
}
int qpow(int a,int b){
	int ans=1;
	while(b){
		if(b&1)ans=(ans*a)%mod;
		a=(a*a)%mod;
		b>>=1;
	}
	return ans;
}
#define endl '\n'
const int INF = 0x3f3f3f3f;
const int N = 1e6+10;
const int maxn = 1e3+10;


int son[N],top[N],f[N],deep[N],sz[N];
struct edge{
	int from,to,next;
}e[N];
int head[N],k;
void add(int u,int v){
	e[++k].from=u;
	e[k].to=v;
	e[k].next=head[u];
	head[u]=k;
}
void dfs1(int u,int fa,int d){//求重儿子
	sz[u]=1,deep[u]=d,f[u]=fa;
	int maxson=-1;
	for(int i=head[u];i;i=e[i].next){
		int v=e[i].to;
		if(v==fa)continue;
		dfs1(v,u,d+1);
		sz[u]+=sz[v];
		if(sz[v]>maxson)maxson=sz[v],son[u]=v;
	}
}
void dfs2(int u,int topfa){//求每个顶点U所在树链的顶端节点
	top[u]=topfa;
	if(!son[u])return ;
	dfs2(son[u],topfa);//重链
	for(int i=head[u];i;i=e[i].next){
		int v=e[i].to;
		if(v!=son[u]&&v!=f[u]){
			dfs2(v,v);//轻链
		}
	}
}
int lca(int x,int y){
	while(top[x]!=top[y]){
		if(deep[top[x]]<deep[top[y]])swap(x,y);
		/*
		  注意比较的是树链顶端的深度,如果比较的x,y的深度会跳过头
		*/
		x=f[top[x]];
	}
	if(deep[x]>deep[y])swap(x,y);
	return x;
}
signed main(){
	std::ios::sync_with_stdio(false);
	cin.tie(0),cout.tie(0);
	int n,q,rt; cin>>n>>q>>rt;
	for(int i=1;i<n;i++){
		int u,v; cin>>u>>v;
		add(u,v); add(v,u);
	}
	dfs1(rt,rt,1);
	dfs2(rt,rt);
	while(q--){
		int x,y; cin>>x>>y;
		cout<<lca(x,y)<<endl;
	}
}

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值