codevs 1036 商务旅行

1036 商务旅行
时间限制: 1 s
空间限制: 128000 KB
题目等级 : 钻石 Diamond
题解
查看运行结果
题目描述 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

思路
裸lca
只需要把要到的城市处理成 查询一对数通过最近公共祖先到达对方的距离（即最短距离）
如要到 1 3 2 5 即查询（1,3）（3,2）（2,5）的lca之和

//小机房的树源码略有改动#include<iostream>#include<cstdio>#include<cstring>#include<algorithm>#include<vector>#include<cmath>using namespace std;const int maxn=50000+50;vector<int>map[maxn],cost[maxn];int deep[maxn],rank[maxn],fa[maxn][25];int n,m,z,a,b;void build(int f,int t,int v){    map[f].push_back(t);    cost[f].push_back(v);    return ;}void done(int f,int t,int v){    deep[t]=deep[f]+1;    fa[t][0]=f;rank[t]=v;    for(int i=0;i<map[t].size();i++){        if(map[t][i]!=f)        done(t,map[t][i],v+cost[t][i]);    }    return ;}void make_lca(){    for(int i=1;i<=log2(n);i++)    for(int j=1;j<=n;j++)    fa[j][i]=fa[fa[j][i-1]][i-1];}int lca(int x,int y){    if(deep[x]<deep[y])     swap(x,y);    for(int i=log2(n);i>=0;i--){        if(deep[fa[x][i]]>=deep[y]) {            x=fa[x][i];        }    }    if(x==y) return x;    for(int j=log2(n);j>=0;j--){        if(fa[x][j]!=fa[y][j]){            x=fa[x][j];y=fa[y][j];        }    }    return fa[x][0];}int main(){    scanf("%d",&n);    int p1,p2,p3;    for(int i=1;i<n;i++){        scanf("%d%d",&p1,&p2);        build(p1,p2,1);        build(p2,p1,1);    }    done(1,1,0);    make_lca();    long long ans=0;    scanf("%d",&m);    for(int i=1;i<=m;i++){        scanf("%d",&a);        if(b) ans+=rank[a]+rank[b]-2*rank[lca(a,b)];        b=a;    }        printf("%lld\n",ans);    return 0;}