Tree[LCA模板]

题目描述

给出一棵带有边权的树,问两点之间的距离。

输入格式

第一行两个整数 n 和 m ,分别表示点数和询问数。
接下来 n-1 行,每行三个整数 x,y,z,表示 x 与 y 通过一条权为 z 的边连接。
接下来 m 行,每行两个整数 x,y,代表一组询问。

##输出格式

输出 m 行,每行一个整数,对应一组询问的答案。

样例数据 1

输入 
3 3
1 2 1
1 3 2
1 2
1 3
2 3
输出
1
2
3

备注

【数据范围】
对 30% 的输入数据 :1≤n,m≤1000
对 100% 的输入数据 :1≤n,m≤100000;1≤z≤10000

代码

dfs_:处理点的深度,距离以及预处理跳 2 i 2^i 步后到达的点
LCA:倍增找公共祖先
注意:LCA_中 f o r for 循环是从 20 0 20-0 不是 20 1 20-1 也不是 0 20 o r 1 20 0-20or1-20

#include <bits/stdc++.h>
using namespace std;
const int N=1e5+50;
int f[N][30],dis[N],tot,dep[N],fir[N];
struct node{
	int to,nxt,w;
}e[N<<1];
inline int read(){
	int cnt=0,f=1;
	char c=getchar();
	while(!isdigit(c)){
		if(c=='-'){
			f=-f;
		}
		c=getchar();
	}
	while(isdigit(c)){
		cnt=(cnt<<3)+(cnt<<1)+(c^48);
		c=getchar();
	}
	return cnt*f;
}
void add(int x,int y,int z){
	e[++tot].nxt=fir[x];fir[x]=tot;e[tot].to=y;e[tot].w=z;
}
void dfs_(int u,int fa){
	dep[u]=dep[fa]+1;
	f[u][0]=fa;
	for(int i=1;i<=20;i++){
		f[u][i]=f[f[u][i-1]][i-1];
	}
	for(int i=fir[u];i;i=e[i].nxt){
		int v=e[i].to;
		if(v==fa)	continue;
		dis[v]=dis[u]+e[i].w;
		dfs_(v,u);
	}
}
int LCA_(int u,int v){
	if(dep[u]<dep[v])	swap(u,v);
	for(int i=20;i>=0;i--){
		if(dep[f[u][i]]>=dep[v])	u=f[u][i];
	}
	if(u==v)	return v;
	for(int i=20;i>=0;i--){
		if(f[u][i]!=f[v][i]){
			u=f[u][i];v=f[v][i];
		}
	}
	return f[u][0];
}
void solve(){
	int n,m;
	n=read(),m=read();
	for(int i=1;i<n;i++){
		int x,y,z;x=read(),y=read(),z=read();
		add(x,y,z);add(y,x,z);
	}
	dfs_(1,0);
	for(int i=1;i<=m;i++){
		int x,y;x=read(),y=read();
		int mid=LCA_(x,y);
		int disn=dis[x]+dis[y]-2*dis[mid];
		printf("%d\n",disn);
	}
}
int main(){
	solve();
	return 0;
}
发布了37 篇原创文章 · 获赞 11 · 访问量 1945

猜你喜欢

转载自blog.csdn.net/weixin_42750325/article/details/102578590