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 longconst 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);}