HDU2586 How far away ？ （树链剖分求LCA）
用树链剖分求LCA的模板；
1 #include<iostream> 2 #include<algorithm> 3 using namespace std; 4 const int maxn=40005; 5 int n,m; 6 int head[maxn],to[maxn<<1],w[maxn<<1],nxt[maxn<<1],tot; 7 int fa[maxn],dep[maxn],son[maxn],top[maxn],size[maxn],dis[maxn]; 8 9 void add(int x,int y,int z){ 10 nxt[++tot]=head[x]; 11 head[x]=tot; 12 to[tot]=y; 13 w[tot]=z; 14 } 15 16 void dfs1(int u,int f){ 17 fa[u]=f,size[u]=1,dep[u]=dep[f]+1; 18 for(int i=head[u];i;i=nxt[i]){ 19 int v=to[i]; 20 if(v==f) continue; 21 dis[v]=dis[u]+w[i]; 22 dfs1(v,u); 23 size[u]+=size[v]; 24 if(size[v]>size[son[u]]) son[u]=v; 25 } 26 } 27 28 void dfs2(int u){ 29 if(u==son[fa[u]]) top[u]=top[fa[u]]; 30 else top[u]=u;//求top数组 31 for(int i=head[u];i;i=nxt[i]){ 32 int v=to[i]; 33 if(v!=fa[u]) dfs2(v); 34 } 35 } 36 37 int LCA(int u,int v){ 38 while(top[u]!=top[v]){ 39 if(dep[top[u]]>dep[top[v]]) u=fa[top[u]]; 40 else v=fa[top[v]]; 41 } 42 return dep[u]>dep[v]?v:u; 43 } 44 45 int main(){ 46 int T; 47 scanf("%d",&T); 48 while(T--){ 49 scanf("%d%d",&n,&m); 50 for(int i=1;i<=n;i++) head[i]=dep[i]=dis[i]=son[i]=0; 51 tot=0; 52 for(int i=1;i<n;i++){ 53 int x,y,z; 54 scanf("%d%d%d",&x,&y,&z); 55 add(x,y,z);add(y,x,z); 56 } 57 dfs1(1,0); 58 dfs2(1); 59 for(int i=1;i<=m;i++){ 60 int x,y; 61 scanf("%d%d",&x,&y); 62 int k=LCA(x,y); 63 printf("%d\n",dis[x]+dis[y]-2*dis[k]); 64 } 65 } 66 return 0; 67 }
