340. 通信线路 二分枚举答案 最小化最大值 最短路里套二分

题目

题解思路

性质

如果存在答案，那么答案必然是某条1到n的路径中边的第k+1大值。且这条路径的第k+1大值最小，（大于答案的边只有k条，即最小）。

二分答案

维护每条路径中最小的大于答案的边的数量。（满足上述性质的最小）
当答案大时，此时大于答案边的数量小于k，因为它不是第k+1大值了。
反之大于。

另外
这题因为取01最短路，可以利用双端队列O(n+m)来手动维护单调性。

AC代码

#include <bits/stdc++.h>
//#include <unordered_map>
//priority_queue
#define PII pair<int,int>
#define ll long long

using namespace std;

const  int  INF =  0x3f3f3f3f;
const  int  N =  1100;
vector <PII> head[N] ; 
int n , m , k ;
int vis[N] ; 
int dis[N] ; 

bool che(int p )
{
    memset(vis,0,sizeof(vis)) ; 
    memset(dis,0x3f,sizeof(dis)) ; 
    priority_queue < PII , vector<PII>, greater<PII> > q ; 
    dis[1] = 0 ; 
    q.push({0,1}) ; 
    while (!q.empty())
    {
        auto sk = q.top() ; 
        q.pop() ;
        if ( vis[sk.second] )
            continue;
        vis[sk.second] = 1 ;
        for (int i = 0 ; i < head[sk.second].size() ; i++ )
        {
            int sp = head[sk.second][i].second ; 
            int pp = head[sk.second][i].first ; 
            int tt = 0 ; 
            if ( pp > p )
                tt = 1 ; 
            if ( !vis[sp] && dis[sp] > dis[sk.second] + tt )
            {
                dis[sp] = dis[sk.second] + tt ; 
                q.push({dis[sp],sp}) ; 
            }
        }
    }
    if ( dis[n] >= k )
        return true ; 
    else
        return false ; 
}
void solve()
{
    cin >> n >> m >> k ;
    for (int i = 1 ; i <= m ; i++ )
    {
        int t1 , t2 , t3 ; 
        cin >> t1 >> t2 >> t3 ; 
        head[t1].push_back({t3,t2}) ; 
        head[t2].push_back({t3,t1}) ; 
    }
    int t1 = 0 , t2 = 1000000 + 1 ;
    while ( t1 < t2 )
    {
        int mid = t1 + t2 >> 1 ; 
        if ( che(mid) ) 
        {
            t2 = mid ; 
        }else
            t1 = mid + 1 ; 
    }
    cout << t2 << "\n" ; 
}
int main()
{
    ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
        solve() ;
    return 0 ;
}