80分求调
查看原帖
80分求调
596437
Raisen_zx楼主2023/8/17 12:18

用的tarjan算法 WA了两个点 ,subtask全TLE,求DALAO帮调

#include<cstdio>
using namespace std;
const int MAXN=500500;
int fa[MAXN],head[MAXN],cnt,hq[MAXN],cq,ans[MAXN],n,m,root;
bool vis[MAXN];
struct e
{
	int to,next,num;
}edge[2*MAXN],query[2*MAXN];
void add(int x,int y)
{
	edge[cnt].to=y;
	edge[cnt].next=head[x];
	head[x]=cnt++;
}
void add_query(int x,int y,int k)
{
	query[cq].to=y;
	query[cq].num=k;
	query[cq].next=hq[x];
	hq[x]=cq++;
}
int find(int x)
{
	if(fa[x]==x) return x;
	return find(fa[x]);
}
void tarjan(int x)
{
	vis[x]=true;
	for(int i=head[x];i!=-1;i=edge[i].next)
	{
		int y=edge[i].to;
		if(!vis[y])
		{
			tarjan(y);
			fa[y]=x;
		}
	}
	for(int i=hq[x];i!=-1;i=query[i].next)
	{
		int y=query[i].to;
		if(vis[y])
		{
			ans[query[i].num]=find(y);
		}
	}
}
int main()
{
	for(int i=0;i<2*MAXN;++i)
	{
		edge[i].next=-1;
		head[i]=-1;
		query[i].next =-1;
		hq[i]=-1;
	}
	scanf("%d %d %d",&n,&m,&root);
	for(int i=1;i<=n;++i)
	{
		fa[i]=i;
	}
	for(int i=1;i<n;++i)
	{
		int u,v;
		scanf("%d %d" ,&u,&v);
		add(u,v);
		add(v,u);
	}
	for(int i=1;i<=m;++i)
	{
		int a,b;
		scanf("%d %d",&a,&b);
		add_query(a,b,i);
		add_query(b,a,i);
	}
	tarjan(root);
	for(int i=1;i<=m;++i)
	{
		printf("%d\n",ans[i]);
	}
	return 0;
}
2023/8/17 12:18
加载中...