dalao的tajan模板

lca:最近公共祖先

求LCA一般有用倍增的和tarjan的,
倍增的是O(nlogn)的但是可以在線;
tarjan是O(n+Q)的但是必須離線。
應當對於不同的題目適當選擇。

tarjan算法的主要思想……是從要求的一對點的訪問過程求來的。
比如以一個點u爲根的多個小子樹內,不同小子樹內的點對的lca都是u。
那麼可以dfs下去,回上去的時候用並查集合並整棵子樹。

看懂網上的或者書上的資料之後代碼其實是很簡潔的……
模板題目網址

#include<bits/stdc++.h>  
using namespace std;  
int read(){  
    int x=0,f=1;char ch=getchar();  
    while (ch<'0' || ch>'9'){if (ch=='-') f=-1;ch=getchar();}  
    while (ch>='0' && ch<='9'){x=x*10+ch-'0';ch=getchar();}  
    return x*f;  
}  
const int       
    N=500005;  
int n,m,S;  
int Qcnt,Ecnt;  
int fa[N],ans[N];  
bool vis[N];  
struct Edge{  
    int next,to;  
}E[N<<1];int head[N];  
struct Query{  
    int next,to,id;  
}Q[N<<1];int qh[N];  
int getfa(int u){  
    if (fa[u]!=u) fa[u]=getfa(fa[u]);  
    return fa[u];  
}  
void add(int u,int v){  
    E[++Ecnt].next=head[u];  
    E[Ecnt].to=v;  
    head[u]=Ecnt;  
}  
void add1(int u,int v,int t){  
    Q[++Qcnt].next=qh[u];  
    Q[Qcnt].to=v;  
    Q[Qcnt].id=t;  
    qh[u]=Qcnt;  
}  
void tarjan(int u){  
    vis[u]=1,fa[u]=u;  
    for (int i=head[u];i;i=E[i].next){  
        int j=E[i].to;  
        if (!vis[j]) tarjan(j),fa[j]=u;  
    }  
    for (int i=qh[u];i;i=Q[i].next){  
        int j=Q[i].to;  
        if (vis[j]) ans[Q[i].id]=getfa(j);  
    }  
}  
int main(){  
    n=read(),m=read(),S=read();  
    int x,y;  
    for (int i=1;i<n;i++){  
        x=read(),y=read();  
        add(x,y),add(y,x);  
    }  
    for (int i=1;i<=m;i++){  
        x=read(),y=read();  
        add1(x,y,i),add1(y,x,i);  
    }  
    tarjan(S);  
    for (int i=1;i<=m;i++)  
        printf("%d\n",ans[i]);  
    return 0;  
}  

%dalao% orz
出處:%dalao%

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章