Tarjan算法求LCA（最近公共祖先）

acwing祖孙询问
此题用倍增在线查询更好，我只是练练Tarjan算法的手，倍增时间复杂度为$O(nlon{g}_n),查询O(lo{g}_n)，tarjan离线lca算法O(n+q)$
Tarjan算法原理:这个博客写的很好

求出每两个点的LCA之后，树上两点的最距离$dis[a][b]=dep[a]+dep[b]-2\ast dep[LCA(a,b)]$,$O(1)$求出

#include<bits/stdc++.h>
using namespace std;
const int N=4e4+10;
typedef pair<int,int> PII;
vector<int>g[N];

struct node
{
    int x;
    int y;
    int z;
};

vector<node>to[N];
int vis[N];
int f[N];
int ans[N];

int find(int x)
{
    if(f[x]==x) return x;
    return f[x]=find(f[x]);
}

void Tarjan(int u,int fa)
{
    vis[u]=1;
    
    for(auto x:g[u])
    {
        if(x==fa) continue;
        if(!vis[x])
        {
            Tarjan(x,u);
            f[x]=u;
        }
    }
    
    for(auto [x,y,z]:to[u])
    {
        if(vis[x]==2)
        {
            int t=find(x);
            if(t==u) // u是x的lca
            {
                if(z==1)
                {
                    ans[y]=1;
                }else ans[y]=2;
            }else ans[y]=0;
        }
    }
    vis[u]=2;//不划分成三个状态也行
}


int main()
{
    int n,m;
    cin>>n;
    for(int i=1;i<N;i++)
    f[i]=i;
    int root=0;
    for(int i=1;i<=n;i++)
    {
        int a,b;
        cin>>a>>b;
        if(b==-1)
        {
            root=a;
        }
        else
        {
            g[a].push_back(b);
            g[b].push_back(a);
        }
    }
    cin>>m;
    for(int i=1;i<=m;i++)
    {
        int a,b;
        cin>>a>>b;
        to[a].push_back({b,i,1});
        to[b].push_back({a,i,2});
    }
    
    Tarjan(root,-1);

    for(int i=1;i<=m;i++)
    {
        cout<<ans[i]<<endl;
    }
    
}