Codevs_P1036 商务旅行(LCA)

题目描述 Description
某首都城市的商人要经常到各城镇去做生意，他们按自己的路线去做，目的是为了更好的节约时间。

假设有N个城镇，首都编号为1，商人从首都出发，其他各城镇之间都有道路连接，任意两个城镇之间如果有直连道路，在他们之间行驶需要花费单位时间。该国公路网络发达，从首都出发能到达任意一个城镇，并且公路网络不会存在环。

你的任务是帮助该商人计算一下他的最短旅行时间。

输入描述 Input Description
输入文件中的第一行有一个整数N，1<=n<=30 000，为城镇的数目。下面N-1行，每行由两个整数a 和b (1<=a, b<=n; a<>b)组成，表示城镇a和城镇b有公路连接。在第N+1行为一个整数M，下面的M行，每行有该商人需要顺次经过的各城镇编号。

输出描述 Output Description
在输出文件中输出该商人旅行的最短时间。

样例输入 Sample Input
5
1 2
1 5
3 5
4 5
4
1
3
2
5
样例输出 Sample Output
7

#include<cstdio>
#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
#define N 30005
int n,m,a,b,step;
vector<int>g[N];
int deep[N];bool v[N];
int f[N][31];
void dfs(int x,int d){
    deep[x]=d;
    for(int i=g[x].size()-1;i>-1;i--){
        if(!v[g[x][i]]){
            f[g[x][i]][0]=x;v[g[x][i]]=true;
            dfs(g[x][i],d+1);
        }
    }
}
void work(){
    for(int i=1;i<=30;i++){
        for(int j=1;j<=n;j++){
            f[j][i]=f[f[j][i-1]][i-1];
        }
    }
}
int lca(int u,int v){
    if(deep[u]<deep[v]) swap(u,v);
    int l=deep[u]-deep[v];
    for(int i=0;i<=30;i++){
        if((1<<i)&l) u=f[u][i];
    }
    if(u==v) return u;
    for(int i=30;i>=0;i--){
        if(f[u][i]!=f[v][i]){
            u=f[u][i],v=f[v][i];
        }
    }
    return f[u][0];
}
int main(){
    scanf("%d",&n);int x,y;
    for(int i=2;i<=n;i++){
        scanf("%d%d",&x,&y);
        g[x].push_back(y);g[y].push_back(x);
    }
    dfs(1,0);work();
    scanf("%d",&m);a=1;
    while(m--){
        scanf("%d",&b);
        if(a==b) continue;
        step=step+deep[a]+deep[b]-2*deep[lca(a,b)];
        a=b;
    }
    printf("%d",step);
}