洛谷 P1948 [USACO08JAN] Telephone Lines S

pystraf

于 2024-07-30 20:29:05 发布

阅读量323

点赞数 5

分类专栏：洛谷题解文章标签： c++ 图论算法数据结构

本文链接：https://blog.csdn.net/sblsf/article/details/140805491

版权

洛谷题解专栏收录该内容

15 篇文章 0 订阅

订阅专栏

Description

给定一张 $n$ 点 $m$ 边的无向图，求一条 $\to n$ 的路径，使得路径上第 $k + 1$ 长的边最短。

Analysis

显然可以发现，答案具有单调性。
又因为题目问的是最大值最小，所以，我们选用二分答案来解决问题。
然后问题就变成了：如果答案为 $mi d$ ，那么 $k$ 最小是多少？（就是最少要经过几条边权 $> mi d$ 的边）
我们可以将边权 $> mi d$ 的边的权看做 $1$ ，边权 $\le mid$ 的看作 $0$ ，此时求新图上一条 $\to n$ 的最短路即可。
由于新图的边权只有 $0$ 和 $1$ ，选用 双端队列 bfs 可以很好地解决这个问题。
贴一下这部分的代码：

// Use `0-1 bfs` (a.k.a `deque bfs`).
// s - Start vertex.
// t - Target vertex.
// mid - The pivot value.
auto bfs = [&](int s, int t, int mid) -> bool {
    // Reset the arrays needed for the computation.
    dis.assign(n, INF);
    vis.assign(n, false);
    // Enqueue the start vertex.
    deque<int> q;
    q.push_back(s);
    dis[s] = 0;
    
    while (q.size()) {
        int u = q.front();
        q.pop_front();
        
        if (vis[u]) continue;
        vis[u] = true;
        
        for (auto &edge: G[u]) {
            int v = edge.first, w = (edge.second > mid);
            if (dis[v] > dis[u] + w) {
                dis[v] = dis[u] + w;
                if (w) q.push_back(v);  // w(u, v) = 1
                else q.push_front(v);   // w(u, v) = 0
            }
        }
    }
    return dis[t] <= k;
};

Code

// Problem: P1948 [USACO08JAN] Telephone Lines S
// Contest: Luogu
// URL: https://www.luogu.com.cn/problem/P1948
// Memory Limit: 125 MB
// Time Limit: 1000 ms
// 
// Powered by CP Editor (https://cpeditor.org)

#include <bits/stdc++.h>
using namespace std;

using i64 = long long;
using PII = pair<int, int>;
const int INF = 1e9;

signed main() {
    ios::sync_with_stdio(0);
    cin.tie(0), cout.tie(0);
    
    int n, m, k;
    cin >> n >> m >> k;
    
    vector<vector<PII>> G(n);
    for (int i = 0, u, v, w; i < m; i++) {
        cin >> u >> v >> w;
        u--, v--;
        
        G[u].emplace_back(v, w);
        G[v].emplace_back(u, w);
    }
    
    vector<int> dis(n, INF);
    vector<bool> vis(n, false);
    
    auto bfs = [&](int s, int t, int mid) -> bool {
        dis.assign(n, INF);
        vis.assign(n, false);
        
        deque<int> q;
        q.push_back(s);
        dis[s] = 0;
        
        while (q.size()) {
            int u = q.front();
            q.pop_front();
            
            if (vis[u]) continue;
            vis[u] = true;
            
            for (auto &edge: G[u]) {
                int v = edge.first, w = (edge.second > mid);
                if (dis[v] > dis[u] + w) {
                    dis[v] = dis[u] + w;
                    if (w) q.push_back(v);
                    else q.push_front(v);
                }
            }
        }
        return dis[t] <= k;
    };
    
    int l = 0, r = 1e6;
    while (l <= r) {
        int mid = l + r >> 1;
        if (bfs(0, n - 1, mid)) r = mid - 1;
        else l = mid + 1;
    }
    cout << (r == 1e6? -1: l) << endl;
    return 0;
}