【模板】LCA

最新推荐文章于 2024-07-20 15:57:21 发布

仰望星空的蚂蚁

最新推荐文章于 2024-07-20 15:57:21 发布

阅读量65

点赞数 1

本文链接：https://blog.csdn.net/cqbzlydd/article/details/108043188

版权

#include<bits/stdc++.h>
using namespace std;
const int N=50010;

struct Edge{
	int v,w;
};

int f[N][20],d[N],dist[N];
vector<Edge> son[N];
int T,n,m,tot,t;
queue<int> q;

void bfs() {
	memset(d,0,sizeof(d));
	q.push(1);d[1]=1;
	while(q.size()) {
		int x=q.front();q.pop();
		for(int i=0;i<son[x].size();i++) {
			int v=son[x][i].v,w=son[x][i].w;
			if(d[v]) continue;
			d[v]=d[x]+1;
			dist[v]=dist[x]+w;
			f[v][0]=x;
			for(int j=1;j<=t;j++) f[v][j]=f[f[v][j-1]][j-1];
			q.push(v);
		}
	}
}

int lca(int x,int y) {
	if(d[x]>d[y]) swap(x,y);
	for(int i=t;i>=0;i--)
	    if(d[f[y][i]]>=d[x]) y=f[y][i];
	if(x==y) return x;
	for(int i=t;i>=0;i--)
	    if(f[x][i]!=f[y][i]) x=f[x][i],y=f[y][i];
	return f[x][0];
}

int main() {
	cin>>T;
	while(T--) {
		cin>>n;
		t=(int)(log(n)/log(2))+1;
		tot=0;
		for(int i=1;i<=n;i++) son[i].clear();
		for(int i=1;i<n;i++) {
			int x,y,z;
			scanf("%d%d%d",&x,&y,&z);
			son[x].push_back((Edge){y,z});
			son[y].push_back((Edge){x,z});
		}
		bfs();
		cin>>m;
	    for(int i=1;i<=m;i++) {
	    	int x,y;
	    	scanf("%d%d",&x,&y);
	    	printf("%d\n",dist[x]+dist[y]-2*dist[lca(x,y)]);
		}
	}
}