1036 商务旅行（LCA例题）

http://codevs.cn/problem/1036/
题目描述 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 <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <vector>
using namespace std;
const int maxn=30000+10;
int pa[maxn],dist[maxn],a[maxn];
vector<int>tree[maxn];
int n,m;
int dfs(int now,int deep)
{
    dist[now]=deep;
    for(int i=0;i<tree[now].size();i++){
        if(dist[tree[now][i]]<0){
            dfs(tree[now][i],deep+1);
        }
    }
}
int camp(int u,int v)
{
    while(dist[u]>dist[v]){
        for(int i=0;i<tree[u].size();i++){
            if(dist[tree[u][i]]<dist[u]){
                u=tree[u][i];
                break;
            }
        }
    }
    return u;
}
int LCA(int u,int v)
{
    int tu=u,tv=v;
    if(dist[u]>dist[v])tu=camp(u,v);
    if(dist[u]<dist[v])tv=camp(v,u);
    while(tu!=tv){
        if(tu!=1)
        for(int i=0;i<tree[tu].size();i++){
            if(dist[tree[tu][i]]<dist[tu]){
                tu=tree[tu][i];
                break;
            }
        }
        if(tv!=1)
        for(int i=0;i<tree[tv].size();i++){
            if(dist[tree[tv][i]]<dist[tv]){
                tv=tree[tv][i];
                break;
            }
        }
    }
    return dist[u]+dist[v]-2*dist[tu];
}
int main()
{
    scanf("%d",&n);
    int u,v;
    for(int i=0;i<n-1;i++){
        scanf("%d%d",&u,&v);
        tree[u].push_back(v);
        tree[v].push_back(u);
    }
    scanf("%d",&m);
    for(int i=0;i<m;i++){
        scanf("%d",&a[i]);
    }
    memset(dist,-1,sizeof(dist));
    dfs(1,0);
    int ans=LCA(1,a[0]);
    for(int i=0;i<m-1;i++)
        ans+=LCA(a[i],a[i+1]);
    printf("%d\n",ans);
    return 0;
}