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

luogu_3379 【模板】最近公共祖先(LCA)

时间:2017-07-01 11:02:47      阅读:246      评论:0      收藏:0      [点我收藏+]

标签:bool   ret   ace   scanf   模板   pac   ext   bsp   can   

 

#include<bits/stdc++.h>

using namespace std;

#define N 500010*2

struct edge{int v,next;}e[N];

struct qedge{int u,v,next,lca,num;}e1[N];

int n,m,t,head[N],top,top1,ans[N],head1[N],p[N];

bool vis[N];

 

int find(int x){return x==p[x]?x:p[x]=find(p[x]);}

void combine(int a,int b){p[find(b)]=find(a);}

 

void add(int u,int v){

    e[++top].v=v; e[top].next=head[u]; head[u]=top;

    e[++top].v=u; e[top].next=head[v]; head[v]=top;

}

 

void qadd(int u,int v,int x){

    e1[++top1].v=v; e1[top1].next=head1[u]; e1[top1].num=x; head1[u]=top1;

    e1[++top1].v=u; e1[top1].next=head1[v]; e1[top1].num=x; head1[v]=top1;

}

 

void tarjin_lca(int u){

    vis[u]=1; p[u]=u;

    for(int i=head[u];i;i=e[i].next){

        int v=e[i].v;

        if(!vis[v]){

            tarjin_lca(v);

            combine(u,v);

        }

    }

    for(int i=head1[u];i;i=e1[i].next){

        int v=e1[i].v;

        if(vis[v]){

            ans[e1[i].num]=find(v);

        }

    }

}

 

int main(){

    scanf("%d%d%d",&n,&m,&t);

    for(int i=1;i<n;i++){

        int u,v;

        scanf("%d%d",&u,&v);

        add(u,v);

    }

    for(int i=1;i<=m;i++){

        int u,v;

        scanf("%d%d",&u,&v);

        qadd(u,v,i);

    }

    tarjin_lca(t);

    for(int i=1;i<=m;i++)printf("%d\n",ans[i]);

    return 0;

}

 

 

luogu_3379 【模板】最近公共祖先(LCA)

标签:bool   ret   ace   scanf   模板   pac   ext   bsp   can   

原文地址:http://www.cnblogs.com/codetogether/p/7101535.html

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