poj3622 二分 双端队列BFS

题意

有向图，k条边免费，除此以外花费为最长边，问最小花费

思路

老题，之前做是分层图最短路裸题，看蓝书新学了个双端队列BFS，用一下。

二分答案，所有比这个答案大的边权为1，否则为0，跑最短路看是否小于等于k即可验证。

在0-1图上最短路，可以跑双端队列BFS，复杂度是线性的。具体做法类似于BFS，只不过是插入的时候，如果权0插头，否则插尾即可。

教训/收获

关于最短路的BFS，如果权全为1，可以直接跑BFS，懒得敲粘SPFA板子也行，复杂度线性，如果是01权，可以跑双端队列BFS，也是线性。如果是正权，跑优先队列BFS，也就是堆优化dij。如果负权，跑SPFA。

代码

#include<cstdio>
#include<iostream>
#include<iomanip>
#include<map>
#include<unordered_map>
#include<string>
#include<queue>
#include<stack>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<cstdlib> 
#include<chrono>
#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define endl "\n"
//#define int long long
//#define double long double
using namespace std;
	typedef long long ll;
	const int maxe=20505;
    const int maxn=1505;
	const int inf=0x3f3f3f3f;
	int n,m,k,q;
    int head[maxn],cnt;
    struct Edge{
        int u,v,w,next;
    }edge[maxe];
    void add(int u,int v,int w=0){
        edge[cnt]={u,v,w,head[u]};
        head[u]=cnt++;
    }
    void addd(int u,int v,int w=0){
        add(u,v,w);
        add(v,u,w);
    }
    void init(){
        memset(head,-1,sizeof head);
        cnt=0;
    }
    int dis[maxn];
    bool vis[maxn];
    deque<int>dq;
    int bfs(int limit){
        memset(dis,0x3f,sizeof dis);
        memset(vis,0,sizeof vis);
        vis[1]=1,dis[1]=0;
        dq.push_back(1);
        while(dq.size()){
            int now=dq.front();dq.pop_front();
            vis[now]=1;
            for(int i=head[now];~i;i=edge[i].next){
                int v=edge[i].v,w=edge[i].w;
                if(vis[v])  continue;
                w=(w>limit?1:0);
                if(w){
                    if(w+dis[now]<dis[v]){
                        dis[v]=w+dis[now];
                        dq.push_back(v);
                    }
                }
                else{
                    if(dis[now]<dis[v]){
                        dis[v]=dis[now];
                        dq.push_front(v);
                    }
                }
            }
        }
        return dis[n];
    }
    bool check(int x){
        return bfs(x)<=k;
    }
	void solve(){
		init();
        cin>>n>>m>>k;
        int ma=0;
        while(m--){
            int u,v,w;
            cin>>u>>v>>w;
            ma=max(ma,w);
            addd(u,v,w);
        }
        if(!check(ma)){
            cout<<-1<<endl;
            return ;
        }
        int l=0,r=ma,mid;
        while(l<r){
            mid=(l+r)/2;
            if(check(mid))  r=mid;
            else            l=mid+1;
        }
        cout<<r<<endl;
	}
	signed main(){
        IOS
		#ifndef ONLINE_JUDGE
		    freopen("IO\\in.txt","r",stdin);
		    freopen("IO\\out.txt","w",stdout);
        #endif
		int tn=1;
		while(tn--){
			solve();
		}
	}