ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

LCA — 欧拉序 + ST表

2021-04-10 21:36:14  阅读:183  来源: 互联网

标签:ch int ST LCA isdigit st 欧拉 getchar


LCA — 欧拉序+ST表

\(O(n\log n)\) 预处理,\(O(1)\) 询问​。

\(lca(x,y)=\) 欧拉序中最早出现的 \(x\) 和 \(y\) 中间深度最浅的点。

Luogu P3379

#include <bits/stdc++.h>
using namespace std;
const int N=500010;
inline int read(){
	int x=0;bool f=false;char ch=getchar();
	while(!isdigit(ch)){
		if(ch=='-')	f=true;
		ch=getchar();
	}
	while(isdigit(ch)){
		x=(x<<3)+(x<<1)+(ch^48);
		ch=getchar();
	}
	return f?-x:x;
}
struct edge{
	int to,next;
}edges[2*N];
int n,m,rt;
int h[N],tot;
int eu[2*N],dep[N],fi[N],cnt;
int st[2*N][30],pos[2*N][30];
void add(int x,int y){
	edges[++tot].next=h[x];
	edges[tot].to=y;
	h[x]=tot;
}
void dfs(int node,int fa,int deep){
	eu[++cnt]=node;
	if(!dep[node]) dep[node]=deep;
	if(!fi[node]) fi[node]=cnt;
	for(int i=h[node];i!=-1;i=edges[i].next){
		int to=edges[i].to;
		if(to!=fa){
			dfs(to,node,deep+1);
			eu[++cnt]=node;
		}
	}
}
void prest(){
	for(int i=1;i<=cnt;i++){
		st[i][0]=dep[eu[i]];
		pos[i][0]=eu[i];
	}
	for(int i=1;(1<<i)<=cnt;i++){
		for(int j=1;(1<<i)+j-1<=cnt;j++){
			if(st[j][i-1]>=st[j+(1<<(i-1))][i-1]){
				st[j][i]=st[j+(1<<(i-1))][i-1];
				pos[j][i]=pos[j+(1<<(i-1))][i-1];
			}else{
				st[j][i]=st[j][i-1];
				pos[j][i]=pos[j][i-1];
			}
		}
	}
}
int lca(int a,int b){
	int x=fi[a],y=fi[b];
	if(x>y) swap(x,y);
	int len=y-x+1;
	int res=log2(len);
	return st[x][res]<=st[y-(1<<res)+1][res]?pos[x][res]:pos[y-(1<<res)+1][res];
}
int main(){
	n=read();m=read();rt=read();
	memset(h,-1,sizeof h);
	for(int i=1;i<=n-1;i++){
		int x,y;
		x=read();y=read();
		add(x,y);
		add(y,x);
	}
	dfs(rt,0,1);
	prest();
	while(m--){
		int a,b;
		a=read();b=read();
		printf("%d\n",lca(a,b));
	}
	return 0;
}

标签:ch,int,ST,LCA,isdigit,st,欧拉,getchar
来源: https://www.cnblogs.com/blueqwq/p/14642160.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有