HDU 5266 pog loves szh III (LCA)

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=5266

题目就是让你求LCA，模版题。注意dfs会栈溢出，所以要扩栈，或者用bfs写。

#pragma comment(linker, "/STACK:102400000,102400000") //扩栈
#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
const int MAXN = 3e5 + 5;
struct data {
    int next , to;
}edge[MAXN << 1];
int par[MAXN][20] , head[MAXN] , cnt , dep[MAXN];

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

void dfs(int u , int d , int p) {
    dep[u] = d;
    par[u][0] = p;
    for(int i = head[u] ; ~i ; i = edge[i].next) {
        int v = edge[i].to;
        if(v == p)
            continue;
        dfs(v , d + 1 , u);
    }
}

void init(int n) {
    for(int k = 0 ; k < 18 ; ++k) {
        for(int i = 1 ; i <= n ; ++i) {
            if(par[i][k] <= 0)
                par[i][k + 1] = par[i][k];
            else
                par[i][k + 1] = par[par[i][k]][k];
        }
    }
}

int lca(int u , int v) {
    if(dep[u] < dep[v]) 
        swap(u , v);
    for(int k = 0 ; k < 20 ; ++k) {
        if(((dep[u] - dep[v]) >> k) & 1) {
            u = par[u][k];
        }
    }
    if(u == v)
        return u;
    for(int k = 19 ; k >= 0 ; --k) {
        if(par[u][k] != par[v][k]) {
            u = par[u][k];
            v = par[v][k];
        }
    }
    return par[u][0];
}

int main()
{
    int n , q , u , v;
    while(~scanf("%d" , &n)) {
        cnt = 0;
        memset(head , -1 , sizeof(head));
        for(int i = 1 ; i < n ; ++i) {
            scanf("%d %d" , &u , &v);
            add(u , v);
            add(v , u);
        }
        dfs(1 , 0 , -1);
        init(n);
        scanf("%d" , &q);
        for(int i = 0; i < q; ++i) {
            scanf("%d %d" , &u , &v);
            printf("%d
" , lca(u , v));
        }
    }
}