Time Limit: 2000/1000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 9359 Accepted Submission(s): 3285
#include<bits/stdc++.h> using namespace std; const int maxn=1e5; struct AdjEdge{ int to,w,next; }adjedges[maxn]; int head[maxn]; int dis[maxn],vset[maxn],dep[maxn],d[maxn][30],first[maxn]; int tot,nn; void init(){ tot=0; nn=0; memset(dep,0,sizeof(dep)); memset(head,-1,sizeof(head)); memset(dis,0,sizeof(dis)); memset(d,0,sizeof(d)); memset(first,0,sizeof(first)); } void addedge(int _u,int _v,int _w){ // adjedges[tot].to=_v; adjedges[tot].w=_w; adjedges[tot].next=head[_u]; head[_u]=tot++; adjedges[tot].to=_u; adjedges[tot].w=_w; adjedges[tot].next=head[_v]; head[_v]=tot++; } void dfs(int _u,int _fa,int _dep){ // printf("%d %d\n",_u,_dep); dep[_u]=_dep; vset[++nn]=_u; first[_u]=nn; for(int i=head[_u];i!=-1;i=adjedges[i].next){ AdjEdge & e = adjedges[i]; if(e.to!=_fa){ dis[e.to]=dis[_u]+e.w; dfs(e.to,_u,_dep+1); vset[++nn]=_u; } } } void ST(){ for(int i=1;i<=nn;i++) d[i][0]=vset[i]; for(int j=1;(1<<j)<=nn;j++){ for(int i=1; i+(1<<j)-1<=nn ; i++){ if(dep[d[i][j-1]]<dep[d[i+(1<<(j-1))][j-1]]) d[i][j]=d[i][j-1]; else d[i][j]=d[i+(1 << (j-1))][j-1]; } } } int RMQ(int L,int R){ int k=0; while((1<<(k+1))<=R-L+1) k++; if(dep[d[L][k]]<=dep[d[R-(1<<k)+1][k]]) return d[L][k]; return d[R-(1<<k)+1][k]; } int main(){ int T,n,q; scanf("%d",&T); while(T--){ init(); int a,b,c; scanf("%d%d",&n,&q); for(int i=1;i<=n-1;i++){ scanf("%d%d%d",&a,&b,&c); addedge(a,b,c); } dfs(1,-1,1); ST(); for(int i=0;i<q;i++){ scanf("%d%d",&a,&b); if(first[a]<=first[b]){ int tmp1=RMQ(first[a],first[b]); printf("%d\n",dis[a]+dis[b]-2*dis[tmp1]); }else{ int tmp1=RMQ(first[b],first[a]); printf("%d\n",dis[a]+dis[b]-2*dis[tmp1]); } } } return 0; }
HDU 2586——How far away ?——————【LCA模板题】
原文:http://www.cnblogs.com/chengsheng/p/4840096.html