蒟蒻80pts倍增求调,wa了两个
查看原帖
蒟蒻80pts倍增求调,wa了两个
780320
xzy_sf楼主2024/7/25 10:44
#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int MAXN=5e5+10;
int n,m,s;
struct Edge{
	int v,nxt;
}edge[2*MAXN];
int head[MAXN],cnt;
void add(int u,int v)
{
	edge[cnt].v=v;
	edge[cnt].nxt=head[u];
	head[u]=cnt;
	cnt++;
}
int lg[MAXN],fa[MAXN][30],dep[MAXN];
void build_lg()
{
	for(int i = 1; i < MAXN; ++i)
		lg[i]=lg[i-1] + (1<<lg[i-1]==i);
}
void dfs(int u,int fath)
{
	fa[u][0]=fath,dep[u]=dep[fath]+1;
	for(int i=1;i<=lg[dep[u]];i++) //
	{
		fa[u][i]=fa[fa[u][i-1]][i-1];
	}
	for(int i=head[u];i;i=edge[i].nxt)
	{
		if(edge[i].v!=fath) dfs(edge[i].v,u);
	}
}
int lca(int x,int y)
{
	if(dep[x]<dep[y]) swap(x,y);
	while(dep[x]>dep[y])
		x=fa[x][lg[dep[x]-dep[y]]-1]; //
	if(x==y) return x;
//	int k=lg[dep[x]]-1;
	for(int k=lg[dep[x]]-1;k>=0;k--)
	{
		if(fa[x][k]==fa[y][k]) continue;
		else x=fa[x][k],y=fa[y][k];
	}
	return fa[x][0];
		
}
int main()
{
	cin>>n>>m>>s;
	for(int i=0;i<n-1;i++)
	{
		int x,y;
		cin>>x>>y;
		add(x,y);
		add(y,x);
	}
	build_lg();
//	cout<<"hello";
	dfs(s,0);
	for(int i=0;i<m;i++)
	{
		int a,b;
		cin>>a>>b;
		cout<<lca(a,b)<<endl;
	}
	return 0;
}

2024/7/25 10:44
加载中...