LCA(模板),Tarjan(离线)算法

https://www.luogu.com.cn/problem/P3379
模板题

点击查看代码

#include <bits/stdc++.h>
#define int long long
using namespace std;

const int N=500010;

int h[N],to[N<<1],ne[N<<1],idx;
int vis[N];
int fa[N];
int ans[N];

struct note//保存询问
{
    int node,id;
};

vector<note> ques[N];

void add(int a,int b)//链式前向星
{
    to[idx]=b,ne[idx]=h[a],h[a]=idx++;
}

int find(int x)//并查集
{
    return fa[x]==x? x:fa[x]=find(fa[x]); 
}

void dfs(int u,int from)
{
    for(int i=h[u];i!=-1;i=ne[i])
    {
        int k=to[i];
        if(k==from)continue;
        dfs(k,u);
        fa[k]=u;
    }
    int len=ques[u].size();
    for(int i=0;i<len;i++)
    {
        if(vis[ques[u][i].node])
        {
            ans[ques[u][i].id]=find(ques[u][i].node);
        }
    }
    vis[u]=1;
}

    signed main()
 {
    memset(h,-1,sizeof h);
    int n,m,s;
    cin>>n>>m>>s;
    for(int i=0;i<=n;i++)fa[i]=i;
    for(int i=1;i<n;i++)
    {
        int a,b;
        cin>>a>>b;
        add(a,b);
        add(b,a);
    }
    for(int i=0;i<m;i++)
    {
        int a,b;
        cin>>a>>b;
        if(a==b)ans[i]=a;//相等时直接保存答案
        note no;
        no.node=b;
        no.id=i;
        ques[a].push_back(no);
        no.node=a;
        no.id=i;
        ques[b].push_back(no);
    }
    dfs(s,0);
    for(int i=0;i<m;i++)
    {
        cout<<ans[i]<<endl;
    }
    return 0;
 }