F. K-th Path

题目链接

题目大意：

求第$K$长的两点最短路。($K<=400$)

思路：

这个$K$的范围很有意思。
在求解最短路的过程中本质都是中转点的存在。（如图$k$作为中转点可以缩短$dis[i][j]$)

当中转点的边权比已存在的第$K$大边权还要大时，中转出的结果也一定比第$K$大要大。故对边权排序，选取前$K$个权值边，跑下多源最短路即可。

Code:

#include <iostream>
#include <map>
#include <set>
#include <queue>
#include <stack>
#include <algorithm>
#include <vector>
#include <string>
#include <iomanip>
#include <cmath>
#include <ctime>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <climits>
//#include <unordered_map>
#define guo312 std::ios::sync_with_stdio(false), cin.tie(0), cout.tie(0)
#define ll long long
#define Inf LONG_LONG_MAX
#define inf INT_MAX
#define endl "\n"
#define PI 3.1415926535898
using namespace std;
const int N1=1e3+10,N2=2e5+10;
map<int,int> mp;
ll n,m,k,cnt=0,dis[N1][N1];

struct PW{
	ll u,v,w;
}a[N2];

bool cmp(PW s1,PW s2){
	return s1.w<s2.w;
}

void init(){
	for(int i=1;i<=cnt;i++){
		for(int j=1;j<=cnt;j++){
			if(i==j) continue;
			if(dis[i][j]==0) dis[i][j]=1e16;
		}
	}
}

void work(){
	for(int i=1;i<=cnt;i++){
		for(int j=1;j<=cnt;j++){
			for(int k=1;k<=cnt;k++){
				dis[j][k]=min(dis[j][k],dis[j][i]+dis[i][k]);
			}
		}
	}
}

int main(){
guo312;
	cin>>n>>m>>k;
	for(int i=1;i<=m;i++){
		cin>>a[i].u>>a[i].v>>a[i].w;
	}
	sort(a+1,a+1+m,cmp);
	for(int i=1;i<=m;i++){
		if(mp[a[i].u]==0) mp[a[i].u]=++cnt;
		if(mp[a[i].v]==0) mp[a[i].v]=++cnt;
		ll u=mp[a[i].u],v=mp[a[i].v],w=a[i].w;
		if(dis[u][v]==0){
			dis[u][v]=dis[v][u]=1e16;
		}
		dis[u][v]=dis[v][u]=min(dis[u][v],w);
		if(i>=k) break;
	}
	init(),work(); vector<ll> ans;
	for(int i=1;i<=cnt;i++){
		for(int j=i+1;j<=cnt;j++){
			if(dis[i][j]==0||dis[i][j]==inf) continue;
			ans.push_back(dis[i][j]);
		}
	}
	sort(ans.begin(),ans.end());
	cout<<ans[k-1];
	return 0;
}