HDU 2586 How far away ？ << LCA转RMQ+ST表 求树上任两点最短距离裸题

 1 #include<bits/stdc++.h>
 2 using namespace std;
 3 const int maxn=4e4+7;
 4 struct Edge{
 5     int to,len;
 6 };
 7 
 8 vector<Edge> G[maxn];
 9 int n;
10 int dfsn[maxn*4];
11 int dist[maxn];
12 int st[maxn];
13 int tot;
14 int dp[maxn*4][20];
15 void init()
16 {
17     for(int i=0;i<=n;i++)
18         G[i].clear();
19     memset(dfsn,0,sizeof(dfsn));
20     memset(dist,0,sizeof(dist));
21     tot=0;
22 }
23 void dfs(int u,int fa)
24 {
25     st[u]=++tot;
26     dfsn[tot]=u;
27     for(int i=0;i<G[u].size();i++)
28     {
29         Edge &v=G[u][i];
30         if(v.to==fa) continue;
31         dist[v.to]=dist[u]+v.len;
32         dfs(v.to,u);
33         dfsn[++tot]=u;
34     }
35 }
36 void rmq()
37 {
38     for(int i=1;i<=tot;i++)
39         dp[i][0]=dfsn[i];
40     for(int j=0;(1<<j)<=tot;j++)
41     {
42         for(int i=1;i+(1<<j)<=tot;i++)
43         {
44             if(dist[dp[i][j]]<dist[dp[i+(1<<j)][j]])
45                 dp[i][j+1]=dp[i][j];
46             else dp[i][j+1]=dp[i+(1<<j)][j];
47         }
48     }
49 }
50 int query(int u,int v)
51 {
52     int ss=u,tt=v;
53     u=st[u],v=st[v];
54     if(u>v) swap(u,v);
55     int len=v-u;
56     int k=0;
57     while((1<<k)<len/2)
58         k++;
59     int LCA;
60     if(dist[dp[u][k]]<dist[dp[v-(1<<k)][k]])
61         LCA=dp[u][k];
62     else LCA=dp[v-(1<<k)][k];
63     return dist[ss]+dist[tt]-2*dist[LCA];
64 }
65 int main()
66 {
67     int T;
68     scanf("%d",&T);
69     while(T--)
70     {
71         int q;
72         scanf("%d%d",&n,&q);
73         init();
74         for(int i=1,u,v,len;i<n;i++)
75         {
76             scanf("%d%d%d",&u,&v,&len);
77             G[u].push_back(Edge {v,len});
78             G[v].push_back(Edge {u,len});
79         }
80         dfs(1,-1);
81         rmq();
82         while(q--)
83         {
84             int u,v;
85             scanf("%d%d",&u,&v);
86             printf("%d\n",query(u,v));
87         }
88     }
89 }