[USACO08OCT] Pasture Walking

Posted 2020-10-26 Neworld2002

题目链接

LCA裸题

#include<cstdio>
#define MAXN 1005
using namespace std;

int N,Q,tot=0;
struct size{
    int v,w,next;
}G[MAXN<<1];
int head[MAXN],anc[11][MAXN],dis[MAXN],d[MAXN];

inline void add(int u,int v,int w){
    G[++tot].v=v;G[tot].w=w;G[tot].next=head[u];head[u]=tot;
}

inline void dfs(int rt,int fa){
    for(register int i=head[rt];i;i=G[i].next){
        int v = G[i].v;if(v==fa)continue;
        dis[v]=dis[rt]+G[i].w;d[v]=d[rt]+1;
        anc[0][v]=rt;
        dfs(v,rt);
    }
}

inline void dp(){
    for(register int i=1;i<=10;++i)
    for(register int j=1;j<=N;++j)
        anc[i][j] = anc[i-1][anc[i-1][j]];
}

inline void swap(int *a,int *b){int t=*a;*a=*b;*b=t;}

inline int LCA(int u,int v){
    if(u==v)return u;
    else if(d[u]<d[v])swap(&u,&v);
    for(register int i=10;i>=0;--i)
        if(d[anc[i][u]]>=d[v])u=anc[i][u];
    if(u==v)return u;
    for(register int i=10;i>=0;--i)
        if(anc[i][u]!=anc[i][v])u=anc[i][u],v=anc[i][v];
    return anc[0][u];
}

int main(){
    

    scanf("%d%d",&N,&Q);
    int u,v,w;
    for(register int i=1;i<N;++i){
        scanf("%d%d%d",&u,&v,&w);
        add(u,v,w);add(v,u,w);
    }
    dis[1]=0;d[1]=0;anc[0][1]=0;
    dfs(1,0);
    dp();
    
    for(register int i=1;i<=Q;++i){
        scanf("%d%d",&u,&v);
        int lca = LCA(u,v);
        printf("%d\n",dis[u]-dis[lca]+dis[v]-dis[lca]);
    }
    return 0;
}