倍增法-lca codevs 1036 商务旅行

 1 #define N 60100
 2 #include<iostream>
 3 using namespace std;
 4 #include<cstdio>
 5 #include<cstring>
 6 #define C 20
 7 typedef long long ll;
 8 int ance[N>>1][C],dis[N>>1],n,a,b,m;
 9 ll sum=0;
10 struct Edge{
11     int v,last;
12 }edge[N<<1];
13 int deep[N>>1],head[N>>1],t=0;
14 void add_edge(int u,int v)
15 {
16     ++t;
17     edge[t].v=v;
18     edge[t].last=head[u];
19     head[u]=t;
20 }
21 void input()
22 {
23     scanf("%d",&n);
24     for(int i=1;i<n;++i)
25     {
26         scanf("%d%d",&a,&b);
27         add_edge(a,b);
28         add_edge(b,a);/*注意树上建边必须是双向的*/
29     }
30 }
31 void dfs(int k)
32 {
33     for(int l=head[k];l;l=edge[l].last)
34     {
35         if(!deep[edge[l].v])
36         {
37             dis[edge[l].v]=dis[k]+1;
38             ance[edge[l].v][0]=k;
39             deep[edge[l].v]=deep[k]+1;
40             dfs(edge[l].v);
41         }
42     }
43 }
44 void chuli_ance()
45 {
46     for(int i=1;(1<<i)<=n;++i)
47       for(int j=1;j<=n;++j)
48       ance[j][i]=ance[ance[j][i-1]][i-1];
49 }
50 int lca(int a,int b)
51 {
52     int i,j;
53     if(deep[a]<deep[b]) swap(a,b);
54     for(i=0;(1<<i)<=deep[a];++i);
55       i--;
56     for(j=i;j>=0;--j)
57       if(deep[a]-(1<<j)>=deep[b])
58       a=ance[a][j];
59     if(a==b) return a;
60     for(j=i;j>=0;--j)
61     {
62         if(ance[a][j]!=-1&&ance[a][j]!=ance[b][j])
63         {
64             a=ance[a][j];
65             b=ance[b][j];
66         }
67     }
68     return ance[a][0];
69 }
70 int main()
71 {
72     input();
73     deep[1]=1;
74     dis[1]=0;
75     memset(ance,-1,sizeof(ance));
76     dfs(1);
77     chuli_ance();
78     scanf("%d",&m);
79     scanf("%d",&a);
80     int sta=a;
81     for(int i=2;i<=m;++i)
82     {
83         scanf("%d",&b);
84         int zuxian=lca(a,b);
85         sum+=(dis[a]+dis[b]-2*dis[zuxian]);
86         a=b;
87     }
88     sum+=dis[sta];
89     cout<<sum<<endl;
90     return 0;
91 }