codeforces 1196 F. K-th Path(第k短路)

http://codeforces.com/contest/1196/problem/F
题意：给出n个点，m条边 求第k短路 k<=400

思路：题解中提到，计算第k短路所需要的边只需要$min（k,m）$即可
大概证明：
1 如果k>m，那么m条边都需要，第k短路必然在，m条边组成的路径中 (废话)
2.k<=m 那么只需要最短的k条路，组成的图中的边即可，因为第k短路必然不会超过单独边的第k大的路(前面组合可能比这个更小)

所以只需要计算出前k条边组成路中最短的第k条即可，把所有需要的边以及相连的顶点全部提取出来，然后离散化，跑一各Floyd算法即可
时间复杂度O(k³）

#include<bits/stdc++.h>
#define fi first
#define se second
#define log2(a) log(n)/log(2)
#define show(a) cout<<a<<endl;
#define show2(a,b) cout<<a<<" "<<b<<endl;
#define show3(a,b,c) cout<<a<<" "<<b<<" "<<c<<endl;
#define tim printf("Time cost : %lf s\n",(double)clock()/CLOCKS_PER_SEC);
using namespace std;
 
typedef long long ll;
typedef long long LL;
typedef pair<int, int> P;
typedef pair<P, ll> LP;
const ll inf = 1e17;
const int N = 2e6 + 10;
const ll mod = 1e18;
const int base = 131;
const double pi = acos ( -1 );
const double eps = 1e-8;
inline ll ksm(ll a,ll b){ll ans=1;while(b){if(b&1)ans=ans*a%mod;a=a*a%mod,b>>=1;}return ans;}
#define a(i,j) a[(i-1)*m+(j)]
#define b(i,j) b[(i-1)*m+(j)]
 
unordered_map<ll, ll> mp;
 
 
 
ll  vis[N],num[N], a[N], b[N],in[N];
ll n, m,  k, x, y, z;
ll cx, cy, cz,cnt, sum, flag, t, ff,w;
vector<ll> v,res;
char s[N];
P p[N];
ll ans;
ll dist[1005][1005];
struct node
{
	ll x,y,w;
 
}ed[N];
bool cmp(node a,node b)
{
	return a.w<b.w;
}
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
 
	cin>>n>>m>>k;
	for(int i=1;i<=m;i++)
	{
		cin>>x>>y>>z;
		ed[++cnt]=node{x,y,z};
	}
	sort(ed+1,ed+cnt+1,cmp);
	for(int i=1;i<=min(m,k);i++)
	{
		v.push_back(ed[i].x);
		v.push_back(ed[i].y);
	}
 
	sort(v.begin(),v.end());
	v.erase(unique(v.begin(),v.end()),v.end());

 
	for(int i=0;i<v.size();i++)
		for(int j=0;j<v.size();j++) dist[i][j]=inf;
 
	for(int i=0;i<v.size();i++) dist[i][i]=0;

 
	for(int i=1;i<=min(m,k);i++)
	{
		
		int x=lower_bound(v.begin(),v.end(),ed[i].x)-v.begin();
		int y=lower_bound(v.begin(),v.end(),ed[i].y)-v.begin();
		
		dist[x][y]=dist[y][x]=min(dist[x][y],ed[i].w);
	}
	//show2("size",v.size())
	for(int k=0;k<v.size();k++)
	{
		for(int i=0;i<v.size();i++)
			for(int j=0;j<v.size();j++)
			dist[i][j]=min(dist[i][j],dist[i][k]+dist[k][j]);
	}
	for(int i=0;i<v.size();i++)
	{
		for(int j=i+1;j<v.size();j++)
			res.push_back(dist[i][j]);
	}
	sort(res.begin(),res.end());
	cout<<res[k-1]<<endl;
 
 
 
 
 
 
}