码迷,mamicode.com
首页 > 其他好文 > 详细

poj 1330 Nearest Common Ancestors (最简单的LCA)

时间:2014-10-04 16:48:16      阅读:146      评论:0      收藏:0      [点我收藏+]

标签:style   blog   color   ar   for   sp   div   c   on   

题意:

给出一棵树的结构。

给出两个点X和Y,求它俩的LCA。

 

思路:

只需求两个点的LCA,用了两种方法,一种离线tarjan,一种直接搞。

看代码。

 

代码:

方法一:直接搞。

int const maxn = 10005;

int T,n,a,b;
int fa[maxn];
int X,Y;

int main(){
    cin>>T;
    while(T--){
        scanf("%d",&n);
        mem(fa,-1);
        rep(i,1,n-1){
            scanf("%d%d",&a,&b);
            fa[b]=a;
        }
        scanf("%d%d",&X,&Y);
        map<int,char> mp;
        int t1=X;
        while(t1!=-1){
            mp[t1]=1;
            t1=fa[t1];
        }
        t1=Y;
        while(t1!=-1){
            if(mp[t1]==1){
                printf("%d\n",t1);
                break;
            }
            t1=fa[t1];
        }
    }
}


 

方法二:离线tarjan。

int const maxn = 10005;

struct node{
    int to,w,next,lca;
};
node edge[maxn*2];

bool vis[maxn];
int cnt1,cnt2;
int head[maxn], fa[maxn];
bool flag;
int X,Y,n;

inline void Addedge(int u,int v,int w){
    edge[++cnt1].w=w;
    edge[cnt1].to=v;
    edge[cnt1].next=head[u];
    head[u]=cnt1;
}

void init(){
    mem(head,-1);
    mem(vis,false);
    cnt1=0;
    flag=false;
    rep(i,1,n) fa[i]=i;
}

int findFa(int x){
    if(fa[x]==x) return fa[x];
    return fa[x]=findFa(fa[x]);
}
void Tarjan_LCA(int u){ //离线LCA算法
    if(flag)
        return;
    fa[u]=u;
    vis[u]=true;
    for(int i=head[u];i!=-1;i=edge[i].next){
        if(!vis[edge[i].to]){
            Tarjan_LCA(edge[i].to);
            fa[edge[i].to]=u;
        }
    }
    if(u==Y && !flag){
        if(vis[X]){
            printf("%d\n",findFa(X));
            flag=true;
            return;
        }
    }
    if(u==X && !flag){
        if(vis[Y]){
            printf("%d\n",findFa(Y));
            flag=true;
            return;
        }
    }
}


int T,a,b;

int main(){
    cin>>T;
    while(T--){
        scanf("%d",&n);
        init();
        bool t1[maxn]={0};
        rep(i,1,n-1){
            scanf("%d%d",&a,&b);
            Addedge(a,b,1);
            t1[b]=true;
        }
        scanf("%d%d",&X,&Y);
        rep(i,1,n) if(!t1[i]){
            Tarjan_LCA(i); //i为树的根结点
            break;
        }
    }
}

 

poj 1330 Nearest Common Ancestors (最简单的LCA)

标签:style   blog   color   ar   for   sp   div   c   on   

原文地址:http://www.cnblogs.com/fish7/p/4006016.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!