1. 程式人生 > >LCA的兩種寫法

LCA的兩種寫法

blank mes col tarjan include pri for namespace continue

第一種是離線的Tarjan算法

#include<cstdio>
using namespace std;
int rd(){
    int x=0,fl=1;char ch=getchar();
    while(ch<0||ch>9){if(ch==-){fl=-1;}ch=getchar();}
    while(ch>=0&&ch<=9){x=(x<<1)+(x<<3)+(ch^48);ch=getchar();}
    return fl*x;
}
int n,m,s,x,y,num1,num2;
int f[500001],dis[500001],a[500001][3],hd1[500001],hd2[500001]; bool vis[500001]; struct star{int nt,to,ds;}e[1000001],e2[1000001]; void add1(int from,int to){ e[++num1].nt=hd1[from]; e[num1].to=to; hd1[from]=num1; } void add2(int from,int to,int ds){ e2[++num2].nt=hd2[from]; e2[num2].to=to; e2[num2].ds=ds; hd2[
from]=num2; } int find(int x){ if(f[x]!=x) f[x]=find(f[x]); return f[x]; } void un(int x,int y){ int ra=find(x),rb=find(y); f[ra]=rb; } void tarjan(int x){ vis[x]=1; for(int i=hd2[x];i;i=e2[i].nt){ int to=e2[i].to,dis=e2[i].ds; if(vis[to]) a[dis][
2]=find(to); } for(int i=hd1[x];i;i=e[i].nt){ int to=e[i].to; if(!vis[to]){ tarjan(to); un(to,x); } } } int main(){ n=rd();m=rd();s=rd(); for(int i=1;i<n;i++){ x=rd();y=rd(); add1(x,y);add1(y,x); } for(int i=1;i<=m;i++){ x=rd();y=rd(); a[i][0]=x;a[i][1]=y; add2(x,y,i);add2(y,x,i); } for(int i=1;i<=n;i++)f[i]=i; tarjan(s); for(int i=1;i<=m;i++) printf("%d\n",a[i][2]); return 0; }

這個代碼跑的飛快但是不太好理解。。

還有一種用倍增的思想

#include<iostream>
#include<cstdio>
using namespace std;
int rd(){
    int x=0,fl=1;char ch=getchar();
    while(ch<0||ch>9){if(ch==-){fl=-1;}ch=getchar();}
    while(ch>=0&&ch<=9){x=(x<<1)+(x<<3)+(ch^48);ch=getchar();}
    return fl*x;
}
int n,m,x,y,s,num=0,hd[500100],dep[500100],f[500100][21];
struct star{int nt,to;}e[1000100];
void add(int fm,int to){e[++num].to=to;e[num].nt=hd[fm];hd[fm]=num;}
void dfs(int x,int fa){
    dep[x]=dep[fa]+1;
    for(int i=1;i<=20;i++)
        f[x][i]=f[f[x][i-1]][i-1];
    for(int i=hd[x];i;i=e[i].nt)
    {
        int to=e[i].to;
        if(to==fa)continue;
        f[to][0]=x;
        dfs(to,x);
    }
}
int lca(int u,int v){
    if(dep[u]<dep[v])swap(u,v);
    for(int i=20;i>=0;i--){
        if(dep[f[u][i]]>=dep[v])
            u=f[u][i];
        if(u==v)return u;
    }
    for(int i=20;i>=0;i--)
        if(f[u][i]!=f[v][i]){
            u=f[u][i];v=f[v][i];
        }
    return f[u][0];
}
void print(int x){
    if(x<0){putchar(-);x=-x;}
    if(x>9)print(x/10);
    putchar(x%10+0);
}
int main(){
    n=rd();m=rd();s=rd();
    for(int i=1;i<n;i++){
        x=rd();y=rd();
        add(x,y);add(y,x);
    }
    dfs(s,0);
    for(int i=1;i<=m;i++){
        x=rd();y=rd();
        print(lca(x,y));
        putchar(\n);
    }
    return 0;
}

emmm....這種比較好理解但是跑的有點慢...

如果不太懂...可以看這個/* */

其實還有一種bfs版的玄學算法也安利一下

/* */

LCA的兩種寫法