记录编号 546641 评测结果 RRRRRRRRRR
题目名称 距离 最终得分 0
用户昵称 Gravatarleon 是否通过 未通过
代码语言 C++ 运行时间 0.004 s
提交时间 2019-11-11 15:54:33 内存使用 15.95 MiB
显示代码纯文本
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<queue>
#include<cstring>
#include<vector>
using namespace std;
const int maxn=20010;
int tot,n,k;
int head[maxn],f[maxn][24],dep[maxn],dis[maxn]; 
struct dd{
	int to,next,v;
}a[maxn];
void add(int x,int y,int z){
	tot++;
	a[tot].to=y;
	a[tot].v=z;
	a[tot].next=head[x];
	head[x]=tot;
}
void dfs(int x,int fa){
	dep[x]=dep[fa]+1;
	for(int i=head[x];i;i=a[i].next){
		int to=a[i].to;
		if(to==fa)continue;
		f[x][1]=fa;
		dis[to]=dis[x]+a[i].v;
		dfs(to,x);
	}
}
int lca(int x,int y){
	if(dep[x]>dep[y])swap(x,y);
	for(int i=24;i>=0;i--){
		if(dep[f[x][i]]<dep[y])x=f[x][i]; 
	} 
	if(x==y)return x;
	for(int i=24;i>=0;i--){
		if(f[x][i]!=f[y][i]){
			x=f[x][i];
			y=f[y][i];
		}
	}
	return f[x][1];
}
int main()
{
	//freopen("heatwave.in","r",stdin);
	//freopen("heatwave.out","w",stdout);
		int x,y,z;
    cin>>n>>k;
	for(int i=1;i<n;i++){
	
		cin>>x>>y>>z;
		add(x,y,z);
		add(y,x,z);
	}
	dep[1]=1;
    dfs(1,0);
    for(int j=1;j<=24;j++){
	for(int i=1;i<=n;i++){
	    f[i][j]=f[f[i][j-1]][j-1];
	}

    for(int i=1;i<=k;i++){
		cin>>x>>y;
		cout<<dis[x]+dis[y]-2*dis[lca(x,y)]; 
    }
	return 0;
}
}