通信线路（二分+思维）

题目描述

在郊区有 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

题目分析

因为该题的最小花费存在单调性，因此我们可以通过二分来找出答案。
然后问题就转化为了check函数怎么写。
那么我们首先要明确check函数需要完成什么事情： 判断图中是否存在一条路径，使得该路径上的所有边权中大于最小花费mid的边数res小于等于k。 而这也是本题需要解决的最关键的问题。
这样的话就需要我们来对这个图做一些处理了，我们可以将图中所有权值大于mid的边的权值改为1，其余的边的权值都改为0，然后跑一遍dijkstra算法，这样最后的结果即为前面的res

代码如下

#include <iostream>
#include <cstdio>
#include <cmath>
#include <string>
#include <cstring>
#include <map>
#include <queue>
#include <vector>
#include <set>
#include <algorithm>
#define LL long long
#define ULL unsigned long long
#define PII pair<int,int>
#define x first
#define y second
using namespace std;
const int N=1e4+5,INF=0x3f3f3f3f;
int n,m,k;
int h[N],e[N*2],ne[N*2],w[N*2],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++;
}
int dijkstra(int val)
{
    memset(st,0,sizeof st);
    memset(dist,0x3f,sizeof dist);
    priority_queue<PII,vector<PII>,greater<PII> >heap;
    dist[1]=0;
    heap.push({0,1});
    while(heap.size())
    {
        PII t=heap.top();
        heap.pop();
        
        int u=t.y;
        if(st[u]) continue;
        st[u]=true;
        
        for(int i=h[u];~i;i=ne[i])
        {
            int v=e[i];
            int x=w[i]>val;		//如果该边的权值大于val(mid)，那么为1，否则为0.
            if(dist[v]>dist[u]+x)
            {
                dist[v]=dist[u]+x;
                heap.push({dist[v],v});
            }
        }
    }
    return dist[n];
}
bool check(int mid)			//判断该结果是否合法
{
    if(dijkstra(mid)>k) return false;
    return true;
}
int main()
{
    memset(h,-1,sizeof h);
    scanf("%d %d %d",&n,&m,&k);
    while(m--)					//建图
    {
        int a,b,c;
        scanf("%d %d %d",&a,&b,&c);
        add(a,b,c);
        add(b,a,c);
    }
    int l=0,r=1e6+1;
    if(!check(r)) {puts("-1"); return 0;}	//先判断是否有解
    while(r>l)				//二分查找答案
    {
        int mid=l+r>>1;
        if(check(mid)) r=mid;
        else l=mid+1;
    }
    printf("%d\n",r);
    return 0;
}