dijkstra+二分

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;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值