poj1330 lca 最近公共祖先问题学习笔记

  1 #include <iostream>
  2 #include <algorithm>
  3 #include <cstdio>
  4 #include <cstdlib>
  5 #include <cstring>
  6 #include <cmath>
  7 #include <ctime>
  8 
  9 using namespace std;
 10 
 11 class Edge
 12 {
 13     public:
 14     int    to;
 15     int    next;
 16 }e[11000];
 17 
 18 int    n,T,cnt;
 19 int    f[11000],depth[11000],anc[11000],p[11000];
 20 bool    visited[11000],In[11000];
 21 
 22 inline    void    Add_edge(const int & x,const int & y)
 23 {
 24     e[++cnt].to=y;
 25     e[cnt].next=p[x];
 26     p[x]=cnt;
 27     return ;
 28 }
 29 
 30 inline    int    get_anc(const int &x)
 31 {
 32     return f[x]==x ? x:f[x]=get_anc(f[x]);
 33 }
 34 
 35 inline    void    Union(const int & x,const int & y)
 36 {
 37     int    S,T;
 38     S=get_anc(x);
 39     T=get_anc(y);
 40     if(S==T)return ;
 41     if(depth[S]<=depth[T])
 42         f[S]=T,depth[S]+=depth[T];
 43     else
 44         f[T]=S,depth[T]+=depth[S];
 45     return ;
 46 }
 47 
 48 void    Dfs(const int &S,const int d)
 49 {
 50     int    i;
 51     depth[S]=d;
 52     for(i=p[S];i;i=e[i].next)
 53     {
 54         Dfs(e[i].to,d+1);
 55     }
 56 
 57     return ;
 58 }
 59 
 60 int    Lca_tarjan(const int & s,const int & t,const int & u)
 61 {
 62     int    i,temp;
 63     
 64     anc[u]=u;
 65     for(i=p[u];i;i=e[i].next)
 66     {
 67         temp=Lca_tarjan(s,t,e[i].to);
 68         if(temp)return temp;
 69         Union(u,e[i].to);
 70         anc[get_anc(u)]=u;
 71     }
 72 
 73     visited[u]=true;
 74     if(s==u&&visited[t])
 75         return anc[get_anc(t)];
 76     if(t==u&&visited[s])
 77         return anc[get_anc(s)];
 78 
 79     return 0;
 80 }
 81 
 82 inline    void    Init()
 83 {
 84     cnt=0;
 85     memset(depth,0,sizeof(depth));
 86     memset(visited,0,sizeof(visited));
 87     memset(anc,0,sizeof(anc));
 88     memset(p,0,sizeof(p));
 89     memset(e,0,sizeof(e));
 90     memset(In,0,sizeof(In));
 91 
 92     for(int i=1;i<=n;++i)
 93     {
 94         f[i]=i;
 95     }
 96 
 97     return ;
 98 }
 99 
100 int main()
101 {
102     //freopen("1330.in","r",stdin);
103     
104     int    i,x,y,s,t,S;
105     
106     scanf("%d",&T);
107     while(T--)
108     {
109         scanf("%d",&n);
110         Init();
111         for(i=1;i<n;++i)
112         {
113             scanf("%d%d",&x,&y);
114             Add_edge(x,y);
115             In[y]=true;
116         }
117         
118         for(S=1;S<=n;++S)
119             if(!In[S])break;
120 
121         scanf("%d%d",&s,&t);
122         Dfs(S,1);
123         printf("%d\n",Lca_tarjan(s,t,S));
124     }
125     
126     return 0;
127 }