[复习]LCA Tree

题目背景
SOURCE：NOIP2015-SHY

题目描述
给出一棵带有边权的树，问两点之间的距离。

输入格式
第一行两个整数 n 和 m ，分别表示点数和询问数。
接下来 n-1 行，每行三个整数 x，y，z，表示 x 与 y 通过一条权为 z 的边连接。
接下来 m 行，每行两个整数 x，y，代表一组询问。

输出格式
输出 m 行，每行一个整数，对应一组询问的答案。

样例数据
输入

3 3
1 2 1
1 3 2
1 2
1 3
2 3

输出

1
2
3

备注
【数据范围】
对 30% 的输入数据 ：
1≤n,m≤1000
对 100% 的输入数据 ：
1≤n,m≤100000；1≤z≤10000

分析： LCA模板

代码：

#include<iostream>#include<cstdio>#include<cstdlib>#include<cstring>#include<string>#include<ctime>#include<cmath>#include<algorithm>#include<cctype>#include<iomanip>#include<queue>#include<set>using namespace std;int getint(){    int sum=0,f=1;    char ch;    for(ch=getchar();!isdigit(ch)&&ch!='-';ch=getchar());    if(ch=='-')    {        f=-1;        ch=getchar();    }    for(;isdigit(ch);ch=getchar())        sum=(sum<<3)+(sum<<1)+ch-48;    return sum*f;}const int N=100010;int n,m,tot;int first[N],nxt[N*2],to[N*2],w[N*2];int fa[N][25],dep[N],dis[N];bool visit[N];void addedge(int x,int y,int z){    tot++;    nxt[tot]=first[x];    first[x]=tot;    to[tot]=y;    w[tot]=z;    tot++;    nxt[tot]=first[y];    first[y]=tot;    to[tot]=x;    w[tot]=z;}void dfs(int u){    for(int p=first[u];p;p=nxt[p])    {        int v=to[p];        if(!visit[v])        {            visit[v]=true;            dep[v]=dep[u]+1;            dis[v]=dis[u]+w[p];            fa[v][0]=u;            for(int i=1;i<=20;++i)                fa[v][i]=fa[fa[v][i-1]][i-1];            dfs(v);        }    }}int lca(int x,int y){    if(dep[x]<dep[y])        swap(x,y);    int len=dep[x]-dep[y];    for(int i=20;i>=0;i--)        if(len&(1<<i))            x=fa[x][i];    if(x==y)        return x;    for(int i=20;i>=0;i--)        if(fa[x][i]!=fa[y][i])        {            x=fa[x][i];            y=fa[y][i];        }    return fa[x][0];}int main(){    //freopen("LCA.in","r",stdin);    //freopen("LCA.out","w",stdout);    int x,y,z;    n=getint(),m=getint();    for(int i=1;i<n;++i)    {        x=getint(),y=getint(),z=getint();        addedge(x,y,z);    }    dep[1]=0,dis[1]=0;    visit[1]=true;    dfs(1);    while(m--)    {        x=getint(),y=getint();        z=lca(x,y);        cout<<dis[x]+dis[y]-2*dis[z]<<'\n';    }    return 0;}

本题结。