通信线路

题目描述

在郊区有 N 座通信基站，P 条 双向 电缆，第 i 条电缆连接基站Ai和Bi。

特别地，1 号基站是通信公司的总站，N 号基站位于一座农场中。

现在，农场主希望对通信线路进行升级，其中升级第 i 条电缆需要花费Li。

电话公司正在举行优惠活动。

农产主可以指定一条从 1 号基站到 N 号基站的路径，并指定路径上不超过 K 条电缆，由电话公司免费提供升级服务。

农场主只需要支付在该路径上剩余的电缆中，升级价格最贵的那条电缆的花费即可。

求至少用多少钱可以完成升级。

输入格式
第1行：三个整数N，P，K。

第2…P+1行：第 i+1 行包含三个整数Ai,Bi,Li。

输出格式
包含一个整数表示最少花费。

若1号基站与N号基站之间不存在路径，则输出”-1”。

数据范围
0≤K<N≤1000,
1≤P≤10000,
1≤Li≤1000000

输入样例：
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

思路

题目要找的是在所有1 -> N的路径中，每条路径当中的第k + 1大边的权值最小值。属于最大值最小问题，所以联想到二分，经过验证也可以很容易发现这道题是可以用二分来进行解决的。

思路来源：https://www.acwing.com yxc

具体做法

二分枚举答案，二分条件为：是否存在1 -> N的路径，路径中边的权值比mid大的条数 <= k。
检测的时候可以将图中大于mid的边的权值看作1，小于等于mid的边的权值看作0，然后就会得到只有0和1权值的图，因此我们可以采用双端队列BFS的方式来进行查找。

代码

#include <cstring>
#include <iostream>
#include <algorithm>
#include <deque>

using namespace std;

const int N = 1010, M = 20010;

int n, m, k;
int h[N], e[M], w[M], ne[M], idx;
int dist[N];
bool st[N];

void add(int a, int b, int c)
{
    e[idx] = b, w[idx] = c, ne[idx] = h[a], h[a] = idx++;
}

//判断是否存在一条路径从1~N，其中权值大于x的边数 <= k
bool check(int x)
{
    memset(dist, 0x3f, sizeof dist);
    memset(st, false, sizeof st);
    dist[1] = 0;
    deque<int> q;
    q.push_front(1);
    
    while (q.size())
    {
        int t = q.front();
        q.pop_front();
        st[t] = true;
        
        for (int i = h[t]; i != -1; i = ne[i])
        {
            int j = e[i];
            if (st[j]) continue;
            int v = (w[i] > x);
            if (dist[j] > dist[t] + v)
            {
                dist[j] = dist[t] + v;
                if (v) q.push_back(j);
                else q.push_front(j);
            }
        }
    }
    
    if (dist[n] <= k) return true;
    return false;
}

int main()
{
    cin >> n >> m >> k;
    memset(h, -1, sizeof h);
    int a, b, c;
    for (int i = 0; i < m; ++i)
    {
        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 (check(mid)) r = mid;
        else l = mid + 1;
    }
    if (l == 1e6 + 1) cout << -1 << endl;
    else cout << l << endl;
    
    return 0;
}