[USACO08OCT] Pasture Walking

题目链接

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;
}