最短路——武

最短路——武

题目描述

其次，Sεlιнα(Selina) 要进行体力比武竞赛。
在 Sεlιнα 所在的城市，有 N个街区，编号为 1∼N，总共有 N-1条的街道连接这些街区， 使得每两个街区之间都直接或间接地有街道将它们相连。Sεlιнα 把通过了文化知识竞赛的参赛男友们召集到她家所在的街区 P ，并以这个街区为起点，让所有参赛男友们向其他街区跑去。这些参赛者们被命令不准重复跑某条街道，而且在规定时间内要尽可能地跑远。比赛结束后，所有参赛者将停留在他们此时所在的街区。之后 Sεlιнα 开始视察结果。现在她知道每个街区都有一些她的参赛男友停留着，她现在想先去看看离她家第 K 近的街区。所以作为一位好帮手，你的任务是要告诉她所有街区中，离 Sεlιнα 家第 K 近的街区与 Sεlιнα 家之间的距离。

输入描述：

第一行三个整数，N,P,K，含义同题面描述。
接下去 N-1行，每行三个整数，u,v,w，表示从第 u 个街区到第 v 个街区有一条权值为w 的街道相连。街区从 1 开始标号。

输出描述：

输出共一行，一个整数，表示所有街区与 Sεlιнα 家所在街区之间最近距离的第 K 小值。

实例1

输入

3 3 2
1 2 4
2 3 5

输出

9

示例2

输入

6 4 3
1 2 7
2 3 2
2 4 2
2 5 10
3 6 3

输出

7

备注：

2≤N≤100000
1≤w≤2000

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<queue>
using namespace std;
const int inf=0x3f3f3f3f;
int n,p,k,d,b,c,dis[100005],vis[100005];
struct node{
	int x;
	int y;
};
vector<node> a[100005];
void dijkstra(int src)
{
	int s;
	queue<int> q;
	vis[src]=1;
	dis[src]=0;
	q.push(src);
	while(!q.empty()){
		s=q.front();
		q.pop();
		vis[s]=1;
		for(int i=0;i<a[s].size();i++){
			if(dis[s]+a[s][i].y<dis[a[s][i].x]){
				dis[a[s][i].x]=dis[s]+a[s][i].y;
				if(!vis[a[s][i].x]){
					q.push(a[s][i].x);
					vis[a[s][i].x]=1;
				}
			}
		}
	}
	return ;
}
int main()
{
	cin>>n>>p>>k;
	memset(dis,0x3f,sizeof(dis));
	memset(vis,0,sizeof(vis));
	node nd;
	for(int i=1;i<n;i++){
		cin>>d>>b>>c;
		nd.x=b;
		nd.y=c;
		a[d].push_back(nd);
		nd.x=d;
		a[b].push_back(nd);
	}
	dijkstra(p);
	sort(dis+1,dis+n+1);
	cout<<dis[k+1]<<endl;
	return 0;
}