acwing:340. 通信线路
在郊区有 N 座通信基站,P 条 双向 电缆,第 i 条电缆连接基站 Ai 和 Bi。
特别地,1 号基站是通信公司的总站,N 号基站位于一座农场中。
现在,农场主希望对通信线路进行升级,其中升级第 i 条电缆需要花费 Li。
电话公司正在举行优惠活动。
农产主可以指定一条从 1 号基站到 N号基站的路径,并指定路径上不超过 K 条电缆,由电话公司免费提供升级服务。
农场主只需要支付在该路径上剩余的电缆中,升级价格最贵的那条电缆的花费即可。
求至少用多少钱可以完成升级。
输入格式
第 11 行:三个整数 N,P,K。
第 2..P+1行:第 i+1 行包含三个整数 Ai,Bi,Li。
输出格式
包含一个整数表示最少花费。
若 1 号基站与 N 号基站之间不存在路径,则输出 −1。
题解:
运用堆优化的dijkstra算法和二分。此题所求即路径上第k+1长的最短边长度,将所有路径求出再查找复杂,这里采用二分。将边长x区间进行二分,将所以大于x的边记为1,小于记为0.再采用dijkstra求最少有几条边大于x,如果d[n]小于k,即x满足条件,如果不满足就增大x,直至找到最小的x满足为第k+1大的边。
代码:
#include<bits/stdc++.h>
using namespace std;
typedef pair<int,int>PII;
const int N=1010,M=10010;
int h[N],ne[M*2],e[M*2],w[M*2],idx;
int d[N];
bool st[N];
int n,m,k;
void add(int a,int b,int c){
e[idx]=b,w[idx]=c,ne[idx]=h[a],h[a]=idx++;
}
bool dijkstra(int W)
{
memset(st,false,sizeof(st));
memset(d,0x3f,sizeof(d));
priority_queue<PII,vector<PII>,greater<PII>> q;
q.push({0,1});
d[1]=0;
while(q.size()){
auto t=q.top();
q.pop();
int v=t.second;
if(st[v]){
continue;
}
st[v]=true;
for(int i=h[v];i!=-1;i=ne[i]){
int j=e[i];
if(d[j]>d[v]+(W<w[i]?1:0)){
d[j]=d[v]+(W<w[i]?1:0);
q.push({d[j],j});
}
}
}
return d[n]<=k;
}
void solve()
{
memset(h,-1,sizeof(h));
cin>>n>>m>>k;
for(int i=1;i<=m;i++){
int a,b,c;
cin>>a>>b>>c;
add(a,b,c);
add(b,a,c);
}
int l=0,r=1e6+1;
while(l<r){
int mid=l+r>>1;
if(dijkstra(mid)){
r=mid;
}else{
l=mid+1;
}
}
if(r!=1e6+1){
cout<<r<<endl;
}else{
cout<<-1<<endl;
}
}
int main()
{
ios::sync_with_stdio(0);
cin.tie(0),cout.tie(0);
int t=1;
while(t--){
solve();
}
return 0;
}