[模板]LCA

洛谷P3379

注意：不能与LCA搞混（打久了就会发现两个还是有很大区别的）

　　　位运算一定要加括号！

　　　for循环从0到logn还是从logn到0看当前的状态更适合哪种

　　　第53行预处理一定要注意！（因为没有下标为-1的数组）

　　　第34行也要注意如何判断当前是否跳点（不需要麻烦的位运算，因为如果能跳，dep[y]自然就会变，如果不跳，dep[y]又不变，每次与(dep[y]-dep[x])进行比较，不影响dep[x]与dep[y]的值；）

 #include<bits/stdc++.h>
 using namespace std;
 #define man 500010
 inline int read()
 {    int x=;bool f=;
     char ch=getchar();
     while(ch<''||ch>''){f=(ch==);ch=getchar();}
     while(ch>=''&&ch<=''){x=(x<<)+(x<<)+(ch^);ch=getchar();}
     return f?(~x+):x;
     }
 int dep[man],f[man][];
 int n,m,logn,root;
 int head[man<<],num=;
 struct edeg
 {    int next,to;}e[man<<];
 inline void add(int from,int to)
 {    e[++num].next=head[from];
     e[num].to=to;
     head[from]=num;
     }
 void dfs(int u,int fa,int depth)
 {    f[u][]=fa;
     dep[u]=depth;
     for(int i=head[u];i;i=e[i].next)
     {    int to=e[i].to;
         if(to==fa) continue;
         dfs(to,u,depth+);
         }
     return ;
     }
 inline int LCA(int x,int y)
 {    if(dep[x]>dep[y]) swap(x,y);
     for(int i=;i<logn;i++)
         if(((dep[y]-dep[x])>>i)&) y=f[y][i];
     if(x==y) return x;
     for(int i=logn;i>=;i--)
     {    if(f[x][i]!=f[y][i])
         {    x=f[x][i];y=f[y][i];
             }
         }
     return f[x][];
     }
 int main()
 {    n=read(),m=read(),root=read();
     logn=(int)(log(n)/log(2.0))+;
     for(int i=;i<n;i++)
     {    int u=read(),v=read();
         add(u,v);add(v,u);
         }
     dfs(root,-,);
     for(int j=;j<logn;j++)
         for(int i=;i<=n;i++)
             if(f[i][j]<) f[i][j+]=-;
             else f[i][j+]=f[f[i][j]][j];
     for(int i=;i<=m;i++)
     {    int x=read(),y=read();
         printf("%d\n",LCA(x,y));
         }
     return ;
     }

 LCA使用BFS应该会快一点吧。。。

 #include<iostream>
 #include<cstdio>
 #include<queue>
 #include<cstring>
 #include<cmath>
 using namespace std;
 #define man 500005
 int n,m,root;
 int head[man<<],num=;
 inline int read()
 {    int x=;bool f=;
     char ch=getchar();
     while(ch<''||ch>''){f=(ch==);ch=getchar();}
     while(ch>=''&&ch<=''){x=(x<<)+(x<<)+(ch^);ch=getchar();}
     return f?(~x+):x;
     }
 struct edge
 {    int next,to;}e[man<<];
 inline void add(int from,int to)
 {    e[++num].next=head[from];
     e[num].to=to;
     head[from]=num;
     }
 int f[man][],dep[man],logn;
 bool vis[man];
 void bfs(int s,int dept)
 {    memset(vis,,sizeof(vis));
     f[s][]=-;dep[s]=dept;
     queue<int>q;
     q.push(s);
     do
     {    int u=q.front();q.pop();vis[u]=;
         for(int i=head[u];i;i=e[i].next)
         {    int to=e[i].to;
             if(!vis[to])
             {    f[to][]=u;
                 dep[to]=dep[u]+;
                 q.push(to);}
                 }
         }while(q.size());
     }
 inline int LCA(int x,int y)
 {    if(dep[x]>dep[y]) swap(x,y);
     for(int i=;i<logn;i++)
     {    if(((dep[y]-dep[x])>>i)&)
              y=f[y][i];
         }
     if(x==y) return x;
     for(int i=logn;i>=;i--)
     {    if(f[x][i]!=f[y][i])
         {    x=f[x][i];
             y=f[y][i];
             }
         }
     return f[x][];
     }
 int main()
 {    n=read(),m=read(),root=read();
     for(int i=;i<n;i++)
     {    int x,y;
         x=read(),y=read();
         add(x,y);add(y,x);
         }
     bfs(root,);
     logn=(int)(log(n)/log(2.0))+;
     for(int j=;j<logn;j++)
         for(int i=;i<=n;i++)
             if(f[i][j]<) f[i][j+]=-;
             else f[i][j+]=f[f[i][j]][j];
     for(int i=;i<=m;i++)
     {    int x,y;
         x=read(),y=read();
         printf("%d\n",LCA(x,y));
         }
     return ;
     }