在郊区有 N 座通信基站,P 条双向电缆,第 i 条电缆连接基站 Ai 和 Bi。特别地,1 号基站是通信公司的总站,N 号基站位于一座农场中。现在,农场主希望对通信线路进行升级,其中升级第 i 条电缆需要花费 Li。
电话公司正在举行优惠活动。农场主可以指定一条从 1 号基站到 N 号基站的路径,并指定路径上不超过 K 条电缆,由电话公司免费提供升级服务。农场主只需要支付在该路径上剩余的电缆中,升级价格最贵的那条电缆的花费即可。求至少用多少钱能完成升级。
一句话题意 在加权无向图上求出一条从 1 号结点到 N 号结点的路径,使路径上第 K+1 大的边权尽量小。
输入格式
第一行三个整数 N,P,K.
接下来 P 行,每行三个整数 Ai,Bi,Li.
输出格式
若不存在从 1 到 N 的路径,输出 -1。否则输出所需最小费用。
样例
Input Output
5 7 1
1 2 5
3 1 4
2 4 8
3 2 3
5 2 9
3 4 7
4 5 6
4
数据范围与提示
0≤K<N≤1000,1≤P≤2000
思路
我们二分最大长度,然后建邻接表时权值比二分值大的就把权值标为1,表示这条线需要公司付账,否则标为0。然后做最短路,如果无法到达就输出− 1 ,直接退出;如果点1 到n要用的长度大于k ,答案就在右边;否则就在左边。
#include<bits/stdc++.h>
#define x first
#define y second
using namespace std;
const int N=1e5+10;
vector<pair<int,int>>v[N];
bool st[N];
int dist[N];
int q[N],w[N],e[N];
int n,m,k,l,r;
int flag;
bool check(){
memset(dist,0x3f,sizeof dist);
dist[1]=0;
st[1]=true;
queue<int>q;
q.push(1);
while(q.size()){
int t=q.front();
q.pop();
st[t]=false;
for(int i=0;i<v[t].size();i++){
int j=v[t][i].x;
if(dist[j]>dist[t]+v[t][i].y){
dist[j]=dist[t]+v[t][i].y;
if(!st[j]){
st[j]=true;
q.push(j);
}
}
}
}
if(dist[n]==0x3f3f3f3f) {
flag=1;
return false;
}
if(dist[n]>k) return false;
return true;
}
int main(){
cin >> n >> m >> k;
for(int i=1;i<=m;i++){
cin >> q[i] >> w[i] >> e[i];
r=max(r,e[i]);
}
while(l<r){
int mid=l+r >> 1;
for(int i=0;i<N;i++){
v[i].clear();
}
memset(dist,0,sizeof dist);
for(int i=1;i<=m;i++){
if(e[i]<=mid){
v[q[i]].push_back({w[i],0});
v[w[i]].push_back({q[i],0});
}
else {
v[q[i]].push_back({w[i],1});
v[w[i]].push_back({q[i],1});
}
}
if(check()){
r=mid;
}
else {
l=mid+1;
}
if(flag==1){
cout << "-1";
return 0;
}
}
cout << l ;
return 0;
}