【bzoj3364】: [Usaco2004 Feb]Distance Queries 距离咨询
Description
奶牛们拒绝跑马拉松，因为她们悠闲的生活无法承受约翰选择的如此长的赛道．因此约翰决心找一条更合理的赛道，他打算咨询你．此题的地图形式与前两题相同．但读入地图之后，会有K个问题．每个问题包括2个整数，就是约翰感兴趣的2个农场的编号，请尽快算出这2个农场间的距离．
Input
第1到I+M行：与前两题相同；
第2+M行：一个整数K(1≤K≤10000)．
第3+M到2+M+K行：每行输入2个整数，代表两个农场．
Output
对每个问题，输出单独的一个整数，给出正确的距离．
倍增模板题。
注意long long RE了好几次。
code：
#include<iostream>
#include<cstdio>
#define int long long
using namespace std;
const int wx=40017;
inline int read(){
int sum=0,f=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch>='0'&&ch<='9'){sum=(sum<<1)+(sum<<3)+ch-'0';ch=getchar();}
return sum*f;
}
int n,m,num;
int head[wx],dep[wx],f[wx][18],dis[wx][18];
struct e{
int nxt,to,dis;
}edge[wx*2];
void add(int from,int to,int dis){
edge[++num].nxt=head[from];
edge[num].to=to;
edge[num].dis=dis;
head[from]=num;
}
void dfs(int u,int fa){
dep[u]=dep[fa]+1;
for(int i=head[u];i;i=edge[i].nxt){
int v=edge[i].to;
if(v==fa)continue;
f[v][0]=u;dis[v][0]=edge[i].dis;
dfs(v,u);
}
}
void pre(){
for(int j=1;j<=17;j++){
for(int i=1;i<=n;i++){
f[i][j]=f[f[i][j-1]][j-1];
dis[i][j]=dis[i][j-1]+dis[f[i][j-1]][j-1];
}
}
}
int LCA(int x,int y){
int re=0;
if(dep[x]<dep[y])swap(x,y);
for(int i=17;i>=0;i--){
if(dep[f[x][i]]>=dep[y]){
re+=dis[x][i];x=f[x][i];
}
}
if(x==y)return re;
for(int i=17;i>=0;i--){
if(f[x][i]!=f[y][i]){
re+=dis[x][i];re+=dis[y][i];
x=f[x][i],y=f[y][i];
}
}
return re+dis[x][0]+dis[y][0];
}
signed main(){
n=read();read();
for(int i=1;i<n;i++){
int x,y,z;
x=read();y=read();z=read();scanf("%s");
add(x,y,z);add(y,x,z);
}
dfs(1,0);pre();
m=read();
for(int i=1;i<=m;i++){
int x,y;
x=read();y=read();
printf("%d
",LCA(x,y));
}
return 0;
}