Transit Tree Path

最新推荐文章于 2020-05-07 15:15:45 发布

To-

最新推荐文章于 2020-05-07 15:15:45 发布

阅读量150

点赞数

分类专栏：数据结构与算法

本文链接：https://blog.csdn.net/dong_qian/article/details/81197596

版权

数据结构与算法专栏收录该内容

70 篇文章 1 订阅

订阅专栏

题目：http://exam.upc.edu.cn/problem.php?cid=1389&pid=3

题意：对给定的一棵树结构，（无向图）问从x到y，途中必须经过点k，问从x到y的最短路程。

因为是树结构，对于任意一点（除了根节点），都只有一个入度，所以从某一点到某一点的最短路是唯一的。

既可以求得从k分到x,y的最短路，再取和即是从x到y的最短路径的长度。

#include<stdio.h>
#include<string.h>
#include<vector>
#include<algorithm>
using namespace std;
vector<pair<long long, long long> > a[200005];
long long  dis[100010]={0};
int vis[100010]={0};
void dfs(int k)
{	
	vis[k]=1;
	for(int i=0;i<a[k].size();i++){
		long long int x=a[k][i].first;
		long long int y=a[k][i].second;
		if(vis[x]==1)
			continue;
		dis[x]=dis[k]+y;
		dfs(x);
	}
}
int main()
{	
	long long int n,m,u,v,w;
	int k;
	scanf("%lld",&n);
	for(int i=1;i<n;i++){
		scanf("%lld %lld %lld",&u,&v,&w);
		a[u].push_back(make_pair(v,w));
		a[v].push_back(make_pair(u,w));
	}
	
	scanf("%lld %d",&m,&k);
	long long int x,y;
	
	dfs(k);
	
	while(m--){
		scanf("%lld %lld",&x,&y);
		
		printf("%lld\n",dis[x]+dis[y]);
		
	}
	
	return 0;
}