BFS(双端队列) + 二分 - Telephone Lines - POJ 3662

BFS(双端队列) + 二分 - Telephone Lines - POJ 3662

在郊区有 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\le K<N\le 1000,1\le P\le 10000,1\le L_i\le 1000000$

Sample Input

5 7 1
1 2 5
3 1 4
2 4 8
3 2 3
5 2 9
3 4 7
4 5 6

Sample Output

4

---

分析：

$给定一个n个点，m条边的无向图，可以从中选择K条边使得边权为0，再求从1号点到N号点的最少花费。$

$两点之间某条路径的花费为该路径上权值最大的边的权值。$

$目标即计算从1号点到N号点的所有路径中，第K+1大的边权的最小值。$

$最好的情况是某条路径上的边的数量不超过K，这样可以把所有边的边权都置0，最终花费就是0。$

$因此我们直接二分答案mid，每次计算从1到N的所有路径中，最少有几条边权大于mid的边，$

$若大于mid的边的数量不超过K，说明mid偏大。否则，说明mid偏小。$

$统计至少有几条边的边权大于mid，可以将大于mid的边的边权赋为1，小于等于mid的边的边权赋为0，$

$然后从1到N，跑一遍最短路，1到N的最短距离就是边权大于mid的边的数量。$

注意：

$①、由于边权仅有0和1两种情况，因此最短路可以用双端队列BFS来实现，时间复杂度是线性的。$

$②、二分的边界：l=0，r=10^6+1。$

$最终答案可取到0，因此左端点为0。$

$最终可能存在不连通的情况，也可能答案取到10^6，为了区分无解的情况，右端点设置为10^6+1。$

代码：

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

using namespace std;

const int N=1010, M=20010;

int n,m,k;
int e[M],ne[M],w[M],h[N],idx;
deque<int> dq;
int dis[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++;
}

bool check(int bound)
{
    memset(st,false,sizeof st);
    memset(dis,0x3f,sizeof dis);
    
    dis[1]=0;
    dq.push_front(1);
    
    while(dq.size())
    {
        int t=dq.front();
        dq.pop_front();
        
        if(st[t]) continue;
        st[t]=true;
        
        for(int i=h[t];~i;i=ne[i])
        {
            int j=e[i], v=w[i]>bound;
            if(dis[j]>dis[t]+v)
            {
                if(v) dq.push_back(j);
                else dq.push_front(j);
                dis[j]=dis[t]+v;
            }
        }
    }
    
    return dis[n]<=k;
}

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(r>1e6) r=-1;
    cout<<r<<endl;
    
    return 0;
}