题解 | #武#

#include <bits/stdc++.h>

using namespace std;

const int N = 1e6+10, M = N*2;

typedef pair<int,int> PII;

int n,p,k;
int e[M],ne[M],h[N],w[M],idx;
int d[N];
bool str[N];

void add(int a,int b,int c)
{
	e[idx]=b,w[idx]=c,ne[idx]=h[a],h[a]=idx++;
}

void dijkstra()
{
	priority_queue<PII,vector<PII>,greater<PII> > heap;
	memset(d,0x3f,sizeof d);
	memset(str,false,sizeof str);
	d[p]=0;
	heap.push({0,p});
	
	while(heap.size())
	{
		PII t=heap.top();
		heap.pop();
		int ver=t.second;
		
		if(str[ver]) continue;
		str[ver]=true;
		
		for(int i=h[ver];i!=-1;i=ne[i])
		{
			int j=e[i];
			if(d[j]>d[ver]+w[i])
			{
				d[j]=d[ver]+w[i];
				heap.push({d[j],j});
			}
		}
	}
}

int main()
{
	cin>>n>>p>>k;
	n--;
	memset(h,-1,sizeof h);
	while(n--)
	{
		int a,b,c;
		cin>>a>>b>>c;
		add(a,b,c),add(b,a,c);
	}
	
	dijkstra();
	sort(d+1,d+idx+1);
	//for(int i=1;i<=3;i++) cout<<d[i]<<" ";
	cout<<d[k+1]<<'\n';
	return 0;
}