题目描述 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
1 #include<iostream> 2 #include<cstdio> 3 using namespace std; 4 5 const int MAXN=100000; 6 struct Edge 7 { 8 int to,next; 9 }E[MAXN]; 10 int node,head[MAXN]; 11 int deep[MAXN],fa[MAXN][20]; 12 bool vis[MAXN]; 13 int n,m,ans; 14 int a[MAXN]; 15 16 void insert(int u,int v) 17 { 18 E[++node]=(Edge){v,head[u]};head[u]=node; 19 E[++node]=(Edge){u,head[v]};head[v]=node; 20 } 21 22 void dfs(int x) 23 { 24 vis[x]=1; 25 for(int i=1;i<=18;i++) 26 { 27 if(deep[x]<(1<<i)) break; 28 fa[x][i]=fa[fa[x][i-1]][i-1]; 29 } 30 for(int i=head[x];i;i=E[i].next) 31 { 32 if(vis[E[i].to]) continue; 33 deep[E[i].to]=deep[x]+1; 34 fa[E[i].to][0]=x; 35 dfs(E[i].to); 36 } 37 } 38 39 int lca(int x,int y) 40 { 41 if(deep[x]<deep[y]) swap(x,y); 42 int d=deep[x]-deep[y]; 43 for(int i=0;i<=18;i++) 44 if((1<<i)&d) x=fa[x][i]; 45 for(int i=18;i>=0;i--) 46 if(fa[x][i]!=fa[y][i]) 47 { 48 x=fa[x][i]; 49 y=fa[y][i]; 50 } 51 if(x==y) return x; 52 else return fa[x][0]; 53 } 54 55 int dis(int x,int y) 56 { 57 int t=lca(x,y); 58 return deep[x]+deep[y]-deep[t]*2; 59 } 60 61 int main() 62 { 63 scanf("%d",&n); 64 for(int i=1;i<n;i++) 65 { 66 int x,y; 67 scanf("%d%d",&x,&y); 68 insert(x,y); 69 } 70 dfs(1); 71 scanf("%d",&m); 72 for(int i=1;i<=m;i++) 73 scanf("%d",&a[i]); 74 for(int i=1;i<m;i++) 75 ans+=dis(a[i],a[i+1]); 76 printf("%d",ans); 77 return 0; 78 }