10分WA+TLE求助QAQ
查看原帖
10分WA+TLE求助QAQ
877156
yyy_Logic楼主2023/8/1 17:06

#include<bits/stdc++.h>
using namespace std;
typedef long double ld;
typedef long long ll;
#define endl '\n'
#define test printf("\ntest\n")
/*·········································*/
const int N = 5e5+10;
int lg[N];


struct node {
	int to,near;
}e[500010<<1];
int head[N],cnt;

//链式前向星
void add(int x,int y)
{
	e[++cnt].to=y;
	e[cnt].near=head[x];
	head[x]=cnt;
}

//预处理
int depth[N],fa[N][22];
void dfs(int now,int fath){
	depth[now]=depth[fath]+1;
	fa[now][0]=fath;
	for(int i=1;i<=lg[depth[now]];i++)
	{
		fa[now][i]=fa[fa[now][i-1]][i-1];
	}
	for(int i=head[now];i;i=e[i].near)
	{
		if(e[i].to!=fath)
			dfs(e[i].to,now);
	}
}

inline int LCA(int ma,int mi){
	if(depth[ma]<depth[mi])
		swap(ma,mi);
	//深度更深的向上跳到与深度更浅的同一级
	while(depth[ma]>depth[mi]){
		ma=fa[ma][depth[ma]-depth[mi]-1];
	}
	if(ma==mi)//如果mi是ma的祖先,那他们的LCA肯定就是mi了
		return ma;
	//两节点一同向上跳
	for(int k=lg[depth[ma]]-1;k>=0;k--)
	{
		if(fa[ma][k]!=fa[mi][k])
		{
			ma=fa[ma][k],mi=fa[mi][k];
		}
	}
	//返回当前两节点其中之一的父节点
	return fa[ma][0];
	
}

void solve()
{
	int n,m,s;
	scanf("%d %d %d",&n,&m,&s);
	for(int i=1;i<=n-1;i++){
		int x,y;
		scanf("%d %d",&x,&y);
		add(x,y);add(y,x);
	}
	//预存lg
	lg[1]=0;
	for(int i=2;i<=n;i++)
		lg[i]=lg[i>>1]+1;
	dfs(s,0);//预处理,s为根节点
	for(int i=1;i<=m;i++){
		int p,q;
		scanf("%d %d",&p,&q);
		printf("%d\n",LCA(p,q));
	}
}
int main()
{
	ios::sync_with_stdio(0);
	cin.tie(0),cout.tie(0);
	int t = 1;
	//	cin>>t;
	while(t--) solve();
	return 0;
}
2023/8/1 17:06
加载中...