LCA及应用

#include<cstdio>
#include<vector>
#include<algorithm>
using namespace std;
#define N 1005
struct Node{int v,w;};
vector<Node> G[N];
int n,m,st[N][12],dep[N],d[N];
void DFS(int u,int p){
    st[u][0]=p;
    for(int j=1;j<12;j++)st[u][j]=st[st[u][j-1]][j-1];
    for(int i=0;i<G[u].size();i++)if(G[u][i].v!=p)
        dep[G[u][i].v]=dep[u]+1,d[G[u][i].v]=d[u]+G[u][i].w, DFS(G[u][i].v,u);
}
int LCA(int u,int v){
    if(dep[u]<dep[v])swap(u,v);
    for(int i=0,d=dep[u]-dep[v];d;i++,d>>=1)if(d&1)u=st[u][i];
    if(u==v)return u;
    for(int i=11;~i;i--)if(st[u][i]!=st[v][i])u=st[u][i],v=st[v][i];
    return st[u][0];
}
int dist(int u,int v){return d[u]+d[v]-2*d[LCA(u,v)];}
int main(){
    scanf("%d%d",&n,&m);
    for(int i=1,u,v,w;i<n;i++)scanf("%d%d%d",&u,&v,&w),G[u].push_back((Node){v,w}),G[v].push_back((Node){u,w});
    DFS(1,0);
    while(m--){
        int a,b;
        scanf("%d%d",&a,&b);
        printf("%d\n",dist(a,b));
    }
    return 0;
}