BZOJ 2953 POI2002 商务旅行

差不多一道LCA模板题目~ 学完tarjan算法 看自己能不能AC掉

 

 1 #include <cstdio>
 2 #include <algorithm>
 3  
 4 int dis[33333],head[33333],fa[33333][33],deep[33333];
 5 int Count,n,q,ans,last=1;
 6  
 7  
 8 struct node{
 9     int v,next,w;
10     node(){}
11     node(int _v,int _next,int _w){
12         v = _v;
13         next = _next;
14         w = _w;
15     }
16 }Edge[233333];
17  
18 void AddEdge(int u,int v,int w){
19     Count++;
20     Edge[Count] = node(v,head[u],w);
21     head[u] = Count;
22 }
23  
24 void dfs(int now,int f){
25     for(int i=head[now];i;i=Edge[i].next){
26         if(f==Edge[i].v) continue;
27         deep[Edge[i].v] = deep[now]+1;
28         fa[Edge[i].v][0]=now;
29         dis[Edge[i].v]=dis[now]+Edge[i].w;
30         dfs(Edge[i].v,now);
31     }
32 }
33  
34 int lca(int x,int y){
35     if(deep[x]<deep[y]) std::swap(x,y);
36     int k = deep[x] - deep[y];
37     for(int i=30;i>=0&&k;i--){
38         if((1<<i)&k){
39             x = fa[x][i];
40             k-=(1<<i);
41         }
42     }
43     if(x==y) return x;
44     for(int i=30;i>=0;i--){
45         if(fa[x][i]!=fa[y][i]){
46             x = fa[x][i];
47             y = fa[y][i];
48         }
49     }
50     if(x!=y) x = fa[x][0];
51     return x;
52 }
53  
54 int main(){
55     //freopen("fuckyou.out","w",stdout);
56     scanf("%d",&n);
57     for(int i=1;i<n;i++){
58         int x,y;
59         scanf("%d%d",&x,&y);
60         AddEdge(x,y,1);
61         AddEdge(y,x,1);
62     }
63     dfs(1,-1);
64     for(int j=1;j<=30;j++)
65         for(int i=1;i<=n;i++)
66             fa[i][j] = fa[fa[i][j-1]][j-1];
67     scanf("%d",&q);
68     while(q--){
69         int x,y;
70         scanf("%d",&x);
71         int fuck = dis[last]+dis[x]-2*dis[lca(last,x)];
72         ans+=fuck;
73         last = x;
74     }
75     printf("%d\n",ans);
76     return 0;
77 }

 

转载于:https://www.cnblogs.com/OIerLYF/p/7496049.html