1. 程式人生 > >樹上倍增 hdu 2586

樹上倍增 hdu 2586

參考部落格:

程式碼:

#include<stdio.h>
#include<iostream>
#include<algorithm>
#include<math.h>
#include<string.h>
using namespace std;
const int max_=8e4+1;
int depth[max_];//節點深度
int gw[max_][25];//第 i 點到2^j父親的距離
int fa[max_][25];//第 i 點的2^j父親
bool vis[max_];//標記陣列
int tot,N;
struct
Tree// { int to; int w; int next; }a[max_*2]; int edge[max_*2]; void add_edge(int x,int y,int w)//建樹 { a[tot].to=y; a[tot].w=w; a[tot].next=edge[x]; edge[x]=tot++; } void dfs(int now,int deep)//dfs預處理 { if(vis[now]==0) { vis[now]=1; depth[now]=deep;//記錄節點的深度
} for(int i=edge[now];i!=-1;i=a[i].next) { int to=a[i].to; if(vis[to]) continue; int w=a[i].w; gw[to][0]=w;//to到父親的距離 fa[to][0]=now;//to的父親節點 dfs(to,deep+1); } } void LCA_init(int n)//預處理 { for(int i=1;i<=n;i++) for(int j=1
;j<=N;j++) { fa[i][j]=fa[fa[i][j-1]][j-1]; gw[i][j]=gw[fa[i][j-1]][j-1]+gw[i][j-1]; } } int LCA_Q(int u,int v) { int ans=0;//距離 if(depth[u]<depth[v]) swap(u,v);//保證大減小 int k; k=depth[u]-depth[v]; for(int i=0;(1<<i)<=k;i++) { if((1<<i)&k) ans+=gw[u][i],u=fa[u][i]; } if(u==v) return ans;//找到了就退出 for(int i=N;i>=0;i--) { if(fa[u][i]!=fa[v][i]) { ans+=gw[u][i], ans+=gw[v][i]; u=fa[u][i], v=fa[v][i]; } } return ans+=gw[v][0]+gw[u][0]; } void init(int n)//初始化 { memset(vis,0,sizeof(vis)); tot=0; N=(int)(log(n+0.0)/log(2)); memset(edge,-1,sizeof(edge)); } int main() { int t; cin>>t; while(t--) { int n,m; scanf("%d %d",&n,&m); init(n); for(int i=0;i<n-1;i++) { int u,v,w; scanf("%d %d %d",&u,&v,&w); add_edge(u,v,w); add_edge(v,u,w); } dfs(1,0); LCA_init(n); while(m--) { int u,v; scanf("%d %d",&u,&v); int L=LCA_Q(u,v); printf("%d\n",L); } /* for(int i=1;i<=n;i++) printf("%d ",fa[i][2]);*/ } } /* 13 2 1 2 1 1 3 1 1 4 1 2 5 1 3 6 1 6 10 1 6 11 1 10 13 1 4 7 1 4 8 1 4 9 1 8 12 1 */ /* for(int x=max0;x>=0;x--) //注意!此處迴圈必須是從大到小!因為我們應該越提越“精確”, if(fa[u][x]!=fa[v][x]) //如果從小到大的話就有可能無法提到正確位置,自己可以多想一下 { u=fa[u][x]; v=fa[v][x]; } return fa[u][0]; //此時u、v的第一個父親就是LCA。*/