POJ 1330（LCA/倍增法模板）

链接：http://poj.org/problem?id=1330

题意：q次询问求两个点u,v的LCA

思路：LCA模板题，首先找一下树的根，然后dfs预处理求LCA（u，v）

AC代码：

#include<iostream>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<set>
#include<string> 
#include<vector>
#include<stack>
#include<queue> 
using namespace std;
typedef long long ll;
const int maxbit = 15;
const int maxn = 1e4+5;
vector<int> G[maxn];
int depth[maxn];
int fa[maxn][maxbit];
int Log[maxn];
int N;
void pre(){
    Log[0] = -1;
    Log[1] = 0,Log[2] = 1;
    for(int i = 3;i<maxn;i++) Log[i] = Log[i/2] + 1;
} 
void dfs(int cur,int father){//dfs预处理 
    depth[cur] = depth[father] + 1;//当前结点的深度为父亲结点+1 
    fa[cur][0] = father;//更新当前结点的父亲结点 
    for(int j = 1;(1<<j)<=N;j++){//倍增更新当前结点的祖先 
        fa[cur][j] = fa[fa[cur][j-1]][j-1];
    }
    for(int i = 0;i<G[cur].size() ;i++){
        if(G[cur][i] != father) {//dfs遍历 
            dfs(G[cur][i],cur);
        }
    }
}
int LCA(int u,int v){
    if(depth[u]<depth[v]) swap(u,v);
    int dist = depth[u] - depth[v];//深度差 
    while(depth[u]!=depth[v]){//把较深的结点u倍增到与v高度相等 
        u = fa[u][Log[depth[u]-depth[v]]];
    }
    if(u == v) return u;//如果u倍增到v，说明v是u的LCA 
    for(int i = Log[depth[u]];i>=0;i--){//否则两者同时向上倍增 
        if(fa[u][i]!=fa[v][i]){//如果向上倍增的祖先不同，说明是可以继续倍增 
            u = fa[u][i];//替换两个结点 
            v = fa[v][i];
        }
    }
    return fa[u][0];//最终结果为u v向上一层就是LCA 
} 
int main()
{
    int t;
    pre();
    scanf("%d",&t);
    while(t--){
        scanf("%d",&N);
        int root ;
        int in[maxn]; 
        for(int i = 0;i<maxn;i++){
            G[i].clear() ;
        }
        memset(in,0,sizeof(in));
        int u,v;
        for(int i = 0;i<N-1;i++){
            scanf("%d%d",&u,&v);
            in[v] = 1;
            G[u].push_back(v);
            G[v].push_back(u);   
        }
        for(int i = 1;i<=N;i++){//寻树的根结点 
            if(in[i] == 0) {
                root = i;
                break;
            }
        }
        dfs(root,0);
        scanf("%d%d",&u,&v);
        int ans = LCA(u,v);
        printf("%d
",ans);
    }
    return 0;
}