1036 商务旅行（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<cstring>
#include<cmath>
#include<iostream>
#include<algorithm>
#include<map>
#include<vector>
#include<queue>
#include<stack>
#define eps 1e-8
#define ll long long
const int inf = 0x3f3f3f3f;
const long long mod=1e9+7;
const int N=35000+20;
using namespace std;
vector<int>q[N];
vector<int>e[N];
int n,m,ans;
int fa[N],sum[N],vis[N];
void init()
{
    memset(vis,0,sizeof(vis));
    for(int i=1;i<=n;i++)
        fa[i]=i;
}
int find(int x)
{
    while(x!=fa[x])
        x=fa[x];
    return x;
}
int dfs(int u,int pre,int val)
{
    sum[u]=val;
    int sz=e[u].size();
    for(int i=0;i<sz;i++)
    {
        int v=e[u][i];
        if(v==pre)
            continue;
        dfs(v,u,val+1);
        fa[v]=u;
    }
    int dsz=q[u].size();
    for(int i=0;i<dsz;i++)
    {
        int v=q[u][i];
        if(vis[v])
        {
            int aim=find(v);
            ans=ans+sum[u]+sum[v]-2*sum[aim];
        }
    }
    vis[u]=1;
}
int main()
{
    int u,v;
    ans=0;
    scanf("%d",&n);
    init();
    for(int i=1;i<n;i++)
    {
        scanf("%d%d",&u,&v);
        e[u].push_back(v);
        e[v].push_back(u);
    }
    scanf("%d",&m);
    u=1;
    for(int i=1;i<=m;i++)
    {
        scanf("%d",&v);
        q[u].push_back(v);
        q[v].push_back(u);
        u=v;
    }
    dfs(1,-1,0);
    printf("%d\n",ans);
}