AcWing 340. 通信线路

题目描述：
在郊区有 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

思路：
二分得到最终答案。
将大于mid的线路的权值记录为1，小于等于的记录为0。
这样，dist数组中记录的就是从起点到终点的大于mid的线路的数量。
如果dist[n]数量大于了要求的k值返回false，反之返回true。
这种就可以锁定最大的一条线缆的长度。
代码：

#include<iostream>
#include<cstring>
#include<queue>
using namespace std;
const int N=1e6+10;
int e[N],ne[N],w[N],h[N],idx;
int dist[N];
bool st[N];
int n,p,k;
void add(int a,int b,int c)
{
    w[idx]=c;
    e[idx]=b;
    ne[idx]=h[a];
    h[a]=idx++;
}
bool check(int k,int mid)
{
    memset(dist,0x3f,sizeof dist);
    memset(st,false,sizeof st);
    deque<int>q;
    q.push_front(1);
    dist[1]=0;
    while(!q.empty())
    {
        int t=q.front();
        q.pop_front();
        if(st[t])
        {
            continue;
        }
        st[t]=true;
        for(int i=h[t];i!=-1;i=ne[i])
        {
            int j=e[i];
            int v=w[i]>mid;
            if(dist[j]>v+dist[t])
            {
                dist[j]=v+dist[t];
                if(v==0)
                {
                    q.push_front(j);
                }
                else
                {
                    q.push_back(j);
                }
            }
        }
    }
    return dist[n]<=k;
}
int main()
{
    memset(h,-1,sizeof h);
    cin>>n>>p>>k;
    for(int i=0;i<p;i++)
    {
        int a,b,c;
        cin>>a>>b>>c;
        add(a,b,c);
        add(b,a,c);
    }
    int l=0;
    int r=1000005;
    while(l<r)
    {
        int mid=(l+r)>>1;
        if(check(k,mid))
        {
            r=mid;
        }
        else
        {
            l=mid+1;
        }
    }
    if(l==1000005)
    {
        cout<<-1<<endl;
        return 0;
    }
    cout<<l<<endl;
}