社区讨论

求调LCA模板

灌水区参与者 2已保存回复 2

讨论操作

快速查看讨论及其快照的属性,并进行相关操作。

当前回复
2 条
当前快照
1 份
快照标识符
@lo2qjxp3
此快照首次捕获于
2023/10/23 18:09
2 年前
此快照最后确认于
2023/10/23 18:09
2 年前
查看原帖
CPP
#include<iostream> 
#include<cstring>
using namespace std;
const int N=5e5+10;
int n,m,s,f[N][30],dep[N],log[N];
int to[N],h[N],ne[N],idx;
bool vst[N]={false};

void add(int u,int v)
{
	to[idx]=v;
	ne[idx]=h[u];
	h[u]=idx++;
}

void dfs(int u)
{
	for(int i=h[u];i!=-1;i=ne[i])
	{
		int v=to[i];
		if(vst[v]) continue;
		vst[v]=true;
		f[v][0]=u;
		dep[v]=dep[u]+1;
		for(int j=1;j<=log[dep[v]];j++)
		{
			f[v][j]=f[f[v][j-1]][j-1];	
		}	
		dfs(v);
	}	
}

int LCA(int x,int y)
{
	if(dep[x]<dep[y]) swap(x,y);
	for(int i=log[dep[x]];i>=0;i--)
	{
		if(dep[f[x][i]]>=dep[y]) x=f[x][i]; 
	}
	if(x==y) return x;
	for(int i=log[dep[x]];i>=0;i--)
	{
		if(f[x][i]!=f[y][i])
		{
			x=f[x][i];
			y=f[y][i];
		}
	}
	return f[x][0];
}

int main()
{
	log[0]=0;
	memset(h,-1,sizeof h);
	cin>>n>>m>>s;
	for(int i=1;i<n;i++)
	{
		int x,y;
		cin>>x>>y;
		add(x,y),add(y,x);
		log[i]=log[i/2]+1;
	}
	dfs(s);
	for(int i=1;i<=m;i++)
	{
		int a,b;
		cin>>a>>b;
		cout<<LCA(a,b)<<" ";
	}
	return 0; 
}


回复

2 条回复,欢迎继续交流。

正在加载回复...