CSP-S 例题17-1 移民火星

描述

2099年，机器人率先入驻火星，收集数据，为人类移民火星打下前站。在有些已经命名的地址例如平乐、西柏坡、文家市、窑店、古绛、胡襄、周庄、郑集、齐都、天柱山、马集等，建立的基站，一些未命名的地址也建立的基站。

目前，火星上总共有 n 个基站，只要机器人能相互到达即可，可以将这些基站视为单独的点。共有 n−1 条道路连接。现在，需要找连接一组三个基站的最短路径。

输入描述

第 1 行一个整数 n，表示 n 个城市区域（1≤n≤50000），城市编号为 0∼n−1。
然后输入 n−1 行数据，每行是 u,v,w，表示 u 区域到 v 区域的 w 长度。
输入 K 组测试数据，每组包括 3 个整数，代表 3 个区域。

输出描述

k 行数据，表示 3 个区域之间的距离。

样例输入 1 

5
0 1 5
0 2 6
1 3 1
1 4 1
2
0 1 2
2 0 3

样例输出 1 

11
12

#include<bits/stdc++.h>
using namespace std;
struct node{
	int v,w,nxt;
}edge[100010];
int n,m,u,v,w;
int cnt,head[50010];
void add(int u,int v,int w){
	cnt++;
	edge[cnt].v=v;
	edge[cnt].w=w;
	edge[cnt].nxt=head[u];
	head[u]=cnt;
}
int p[50010][20],dis[50010],dep[50010];
void dfs(int u,int fa){
	dep[u]=dep[fa]+1;
	p[u][0]=fa;
	for(int i=head[u];i;i=edge[i].nxt){
		int v=edge[i].v;
		int w=edge[i].w;
		if(v==fa) continue;
		dis[v]=dis[u]+w;
		dfs(v,u);
	}
}
int lca(int x,int y){
	if(dep[x]<dep[y]){
		swap(x,y);
	}
	for(int i=19;i>=0;i--){
		if(dep[x]-(1<<i) >= dep[y]){
			x=p[x][i];
		}
	}
	if(x==y) return x;
	for(int i=19;i>=0;i--){
		if(p[x][i]!=p[y][i]){
			x=p[x][i];
			y=p[y][i];
		}
	}
	return p[x][0];
}
int main(){
	cin>>n;
	for(int i=1;i<n;i++){
		cin>>u>>v>>w;
		u++;
		v++;
		add(u,v,w);
		add(v,u,w);
	}
	dfs(1,0);
	for(int j=1;j<=19;j++){
		for(int i=1;i<=n;i++){
			p[i][j]=p[p[i][j-1]][j-1];
		} 
	}
	cin>>m;
	while(m--){
		int a,b,c;
		cin>>a>>b>>c;
		a++;
		b++;
		c++;
		int ab=dis[a]+dis[b]-2*dis[lca(a,b)];
		int ac=dis[a]+dis[c]-2*dis[lca(a,c)];
		int bc=dis[b]+dis[c]-2*dis[lca(b,c)];
		cout<<(ab+bc+ac)/2<<endl;
	}
	return 0;
}