【模板】Lca倍增法

Codevs 1036 商务旅行

#include<cstdio>
#include<cmath>
#include<algorithm>
using namespace std;
const int maxn=100010;
int n,m,x,y,ans=0,tot=0,last[maxn],dep[maxn],p[maxn][32];
struct edge{
    int to,pre;
}e[maxn<<1];

void read(int &k){
    k=0; int f=1; char c=getchar();
    while (c<'0'||c>'9')c=='-'&&(f=-1),c=getchar();
    while ('0'<=c&&c<='9')k=k*10+c-'0',c=getchar();
    k*=f;
}
void add(int x,int y){
    e[++tot].to=y;
    e[tot].pre=last[x];
    last[x]=tot;
}
void dfs(int u,int fa){
    dep[u]=dep[fa]+1; p[u][0]=fa;
    for (int i=1;i<=log2(dep[u]);i++) p[u][i]=p[p[u][i-1]][i-1];
    for (int i=last[u],to=e[i].to;i;i=e[i].pre,to=e[i].to)
    if (to!=fa) dfs(to,u);
}
int lca(int x,int y){
    if (dep[x]<dep[y]) swap(x,y);
    if (dep[x]>dep[y])
    for (int i=log2(dep[x]-dep[y]+1);i>=0,dep[x]!=dep[y];i--)
        if (dep[p[x][i]]>=dep[y]) x=p[x][i];
     if (x==y) return x;
     for (int i=log2(dep[x]);i>=0;i--)
     if (p[x][i]!=p[y][i]) x=p[x][i],y=p[y][i];
     return p[x][0];
}
int main(){
    read(n);
    for (int i=1;i<n;i++){
        read(x); read(y);
        add(x,y); add(y,x);
    }
    dfs(1,0);
    read(m); read(x);
    for (int i=1;i<m;i++){
        read(y);
        ans+=dep[x]+dep[y]-(dep[lca(x,y)]<<1);
        x=y;
    }
    printf("%d
",ans);
    return 0;
}