【codevs】2370 小机房的树

 1 #include<cstdio>
 2 #include<algorithm>
 3 using namespace std;
 4 const int maxn=50010;
 5 struct point{int from,to,v;}e[maxn*2];
 6 int n,m,cnt,a,b,c;
 7 int head[maxn],deep[maxn],x[maxn][17],di[maxn];
 8 //deep数组表示节点所在的层数，di数组表示节点到根节点的距离 
 9 //x[i][j]的含义为：节点i的祖先中与其相差2^j层的节点；例如，x[i][0]代表节点i的父亲 
10 bool f[maxn];//判断节点是否访问过 
11 void add(int a,int b,int c)//邻接表建边 
12 {cnt++;e[cnt].from=head[a];e[cnt].to=b;e[cnt].v=c;head[a]=cnt;}
13 void dfs(int k)
14 {
15     f[k]=true;
16     for(int i=1;(1<<i)<=deep[k];i++)
17         x[k][i]=x[x[k][i-1]][i-1];//递推得出x数组 
18     for(int i=head[k];i;i=e[i].from) 
19     {
20         if(!f[e[i].to])
21         {
22             x[e[i].to][0]=k;//将节点k标记为e[i].to的父亲 
23             deep[e[i].to]=deep[k]+1;//层数+1 
24             di[e[i].to]=di[k]+e[i].v;
25             dfs(e[i].to);
26         }
27     }
28 }
29 int solve(int ri,int rj)
30 {
31     if(deep[ri]<deep[rj])swap(ri,rj);
32     int d=deep[ri]-deep[rj]; 
33     for(int i=0;(1<<i)<=d;i++)//利用倍增思想，将ri节点跳到与rj节点同层 
34         if((1<<i)&d)ri=x[ri][i];
35     if(ri==rj)return ri;//找到最近公共祖先 
36     for(int i=16;i>=0;i--)
37         if((1<<i)<=deep[rj]&&x[ri][i]!=x[rj][i])
38             ri=x[ri][i],rj=x[rj][i];//同时往上跳，直到ri与rj为兄弟节点为止 
39     return x[ri][0];//返回他们的父亲 
40 }
41 int main()
42 {
43     scanf("%d",&n);
44     for(int i=1;i<n;i++)
45     {
46         scanf("%d%d%d",&a,&b,&c);
47         add(a,b,c);
48         add(b,a,c);
49     }
50     dfs(0);//此处假设以0为根节点 
51     scanf("%d",&m);
52     while(m--)
53     {
54         scanf("%d%d",&a,&b);
55         printf("%d\n",di[a]+di[b]-2*di[solve(a,b)]);
56     }
57     return 0;
58 }