求助—之前的lca板子被卡了#11和#12
查看原帖
求助—之前的lca板子被卡了#11和#12
159959
虫洞吞噬者楼主2021/11/3 11:41

RT,所以很想知道#11和#12构造了什么数据来卡lca,会卡什么情况

code:

#include<cmath>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int n,m,cnt,ans,s;
int head[500100],vis[500100],dep[500100],fa[500100],siz[500100];
int f[500100][50];
struct node{
	int nxt,to;
}edge[1000200];
void add(int from,int to)
{
	edge[++cnt].nxt=head[from];
	edge[cnt].to=to;
	head[from]=cnt;
}
void dfs(int pre,int cur)
{
	f[cur][0]=pre;
	dep[cur]=dep[pre]+1;
	for(int i=head[cur];i;i=edge[i].nxt)
	{
		int nxt=edge[i].to;
		if(nxt!=pre)dfs(cur,nxt);
	}
}
void init()
{
	for(int i=1;(1<<i)<=n;++i)
		for(register int j=1;j<=n;++j)
			f[j][i]=f[f[j][i-1]][i-1];		
}
int lca(int x,int y)
{
	if(dep[x]<dep[y])swap(x,y);
//	int flag=0,ans;
	int t=dep[x]-dep[y];
	for(int j=int(log(n));j>=0;--j)
		if((t>>j)&1)x=f[x][j];
	if(x==y)return x;
	for(int i=int(log(n));i>=0;--i)
	{
		if(f[x][i]!=f[y][i])
		{
			x=f[x][i];
			y=f[y][i];
		}
	}
	return f[x][0];
}
int main()
{
	scanf("%d%d%d",&n,&m,&s);
	for(int i=1;i<n;++i)
	{
		int a,b;
		scanf("%d%d",&a,&b);
		add(a,b);
		add(b,a);
	}
	for(int i=1;i<=n;++i)f[i][0]=edge[i].to;
	dfs(0,s);
	init();
	for(int i=1;i<=m;++i)
	{
		int a,b;
		scanf("%d%d",&a,&b);
		if(a==b)printf("%d\n",a);
		else printf("%d\n",lca(a,b));
	}
	return 0;
}
2021/11/3 11:41
加载中...