Problem: [Poi0202]Travelling Salesman 最近公共祖先

Problem: [Poi0202]Travelling Salesman 最近公共祖先

Time Limit: 1 Sec Memory Limit: 128 MB

Description

某首都城市的商人要经常到各城镇去做生意，他们按自己的路线去做，目的是为了更好的节约时间。
假设有N个城镇，首都编号为1，商人从首都出发，其他各城镇之间都有道路连接，任意两个城镇之间如果有直连道路
在他们之间行驶需要花费单位时间。该国公路网络发达，从首都出发能到达任意一个城镇，并且公路网络不会存在环。
你的任务是帮助该商人计算一下他的最短旅行时间。

Input

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

Output

输出该商人旅行的最短时间。

Sample Input

5
1 2
1 5
3 5
4 5
4
1
3
2
5

Sample Output

7
代码如下

#include<cstdio>
#include<cstring>
using namespace std;
#define elif else if
int g[30010][510],dis[30010],num[30010],point[30010],n,m,ans=0;
int fa[30010],que[30010][3];
bool bo[30010];
void buildtree(int u) {
    for (int i=1; i<=num[u]; i++)
        if (!bo[g[u][i]]) {
            bo[g[u][i]]=true;
            dis[g[u][i]]=dis[u]+1;
            buildtree(g[u][i]);
        }
}
int find(int u) {
    return fa[u]==u?u:fa[u]=find(fa[u]);
}
void tarjan(int u) {
    bo[u]=true;
    for (int i=1; i<=que[u][0]; i++)
        if (bo[que[u][i]]) ans+=dis[u]+dis[que[u][i]]-2*dis[find(que[u][i])];
    for (int i=1; i<=num[u]; i++)
        if (!bo[g[u][i]]) {
            tarjan(g[u][i]);
            fa[g[u][i]]=u;
        }
}
int main() {
    scanf("%d",&n);
    for (int i=1; i<=n; i++) fa[i]=i;
    for (int i=1; i<n; i++) {
        int x,y;
        scanf("%d%d",&x,&y);
        g[x][++num[x]]=y;
        g[y][++num[y]]=x;
    }
    bo[1]=true;
    dis[1]=0;                                                  
    buildtree(1);
    scanf("%d",&m);
    memset(bo,0,sizeof(bo));
    for (int i=1; i<=m; i++) {
        scanf("%d",&point[i]);
        if (i==1&&point[i]==1) continue;
        elif (i==1&&point[i]!=1) {
            que[point[i]][++que[point[i]][0]]=1;
            que[1][++que[1][0]]=point[i];
        } else {
            que[point[i]][++que[point[i]][0]]=point[i-1];
            que[point[i-1]][++que[point[i-1]][0]]=point[i];
        }
    }
    tarjan(1);
    printf("%d",ans);
    return 0;
}