用的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;
}