【wikioi】1036商務旅行

題目描述  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出發，而後依次求每兩個點之間的lca，而後過程當中把deep加起來。。話說這道題被坑了很久。。從下午到晚上。。而後發現是無向邊，因而要加雙向邊，果斷推翻重寫。。而後終於AC了。。

【代碼】：

 1 #include<cstdio>
 2 #include<algorithm>
 3 
 4 using namespace std;
 5 
 6 struct edge{
 7     int u,v,next;
 8 }e[30000 * 2 + 5];
 9 int head[37000];
10 int deep[37000];
11 int p[37000][37];
12 int n,m,tot,k=1;
13 
14 void adde(int u,int v,int k)
15 {
16     e[k].next=head[u];
17     head[u]=k;
18     e[k].u=u;
19     e[k].v=v;
20 }
21 void dfs(int fa,int x)
22 {
23     if(deep[x]) return ;
24     deep[x]=deep[fa]+1;
25     p[x][0]=fa;
26     for(int i=1;p[p[x][i-1]][i-1];i++) {
27         p[x][i]=p[p[x][i-1]][i-1];
28     }
29     for(int i=head[x];i;i=e[i].next) {
30         dfs(x,e[i].v);
31     }
32 }
33 int lca(int u,int v)
34 {
35     if(deep[u]<deep[v]) swap(u,v);
36     if(deep[u]>deep[v]) {
37         int dva=deep[u]-deep[v];
38         for(int i=0;i<=24;i++) {
39             if(dva&(1<<i)) {
40                 u=p[u][i];
41             }
42         }
43     }
44     if(u!=v) {
45         for(int i=24;i>=0;i--) {
46             if(p[u][i]!=p[v][i]) {
47                 u=p[u][i];
48                 v=p[v][i];
49             }
50         }
51     }
52     if(u==v) return u;
53     else return p[u][0];
54 }
55 int main()
56 {
57     scanf("%d",&n);
58     for(int i=1;i<n;i++) {
59         int u,v;
60         scanf("%d%d",&u,&v);
61         adde(u,v,k++);
62         adde(v,u,k++);
63     }
64     deep[1]=1;
65     for(int i=head[1];i;i=e[i].next) {
66         dfs(1,e[i].v);
67     }
68     int a = 1;
69     scanf("%d",&m);
70     for(int i=1;i<=m;i++) {
71         int b;
72         scanf("%d",&b);
73         int Lca=lca(a,b);
74         tot+=deep[a]+deep[b]-2*deep[Lca];
75         a=b;
76     }
77     printf("%d",tot);
78     return 0;
79 }

View Code