BZOJ 1602 牧场行走

直接写一波Lca就好了

#include<cstdio>
#include<cmath>
#include<algorithm>
using namespace std;
const int maxn=1010;
int n,m,x,y,z,ans=0,tot=0,dis[maxn],last[maxn],dep[maxn],p[maxn][32];
struct edge{int to,pre,dis;}e[maxn<<1];

void read(int &k){
    k=0; int f=1; char c=getchar();
    while (c<'0'||c>'9')c=='-'&&(f=-1),c=getchar();
    while ('0'<=c&&c<='9')k=k*10+c-'0',c=getchar();
    k*=f;
}
void add(int x,int y,int z){e[++tot].to=y;e[tot].pre=last[x];e[tot].dis=z;last[x]=tot;}
void dfs(int u,int fa){
    dep[u]=dep[fa]+1; p[u][0]=fa;
    for (int i=1;i<=log2(dep[u]);i++) p[u][i]=p[p[u][i-1]][i-1];
    for (int i=last[u],to=e[i].to;i;i=e[i].pre,to=e[i].to)
    if (to!=fa) dis[to]=dis[u]+e[i].dis,dfs(to,u);
}
int lca(int x,int y){
    if (dep[x]<dep[y]) swap(x,y);
    if (dep[x]>dep[y])
    for (int i=log2(dep[x]-dep[y]+1);i>=0,dep[x]!=dep[y];i--)
        if (dep[p[x][i]]>=dep[y]) x=p[x][i];
     if (x==y) return x;
     for (int i=log2(dep[x]);i>=0;i--)
     if (p[x][i]!=p[y][i]) x=p[x][i],y=p[y][i];
     return p[x][0];
}
int main(){
    read(n); read(m);
    for (int i=1;i<n;i++){
        read(x); read(y); read(z);
        add(x,y,z); add(y,x,z);
    }
    dfs(1,0);
    for (int i=1;i<=m;i++) read(x),read(y),printf("%d
",dis[x]+dis[y]-2*dis[lca(x,y)]);
    return 0;
}