Loading...

洛谷P1951 收费站_NOI导刊2009提高(2)

check评论:0 条 remove_red_eye浏览量:417 change_historyTags:编程学习笔记
作者 : deco date_range日期 : 2018-10-24

一道二分的题,我们每次二分最小费用即可,这个没什么好说的

这道题直接$spfa$会被卡成$70$,我们需要使用$SLF$优化

我们用$deque$来实现,如果当前队列非空并且队首元素距离较小,我们就将这个元素$push\_back()$,否则就$push\_front()$

#include <bits/stdc++.h>
using namespace std;
#define int long long
int cost[10010],head[10010],dis[10010],vis[10010],cnt,n,m,maxn,s,t,flow;
struct edge
{
    int next,to,dis;
}e[200020];
void add(int u,int v,int dis)
{
    cnt++;
    e[cnt].to=v;
    e[cnt].next=head[u];
    e[cnt].dis=dis;
    head[u]=cnt;
}
bool spfa(int maxm)
{
    deque<int> q;
    memset(dis,0x3f,sizeof(dis));
    memset(vis,0,sizeof(vis));
    q.push_front(s); 
    vis[s]=1;
    dis[s]=0;
    while(!q.empty())
    {
        int u=q.front();
        q.pop_front();
        vis[u]=0;
        if(cost[u]>maxm)
        {
            continue;
        }
        for(int i=head[u];i;i=e[i].next)
        {
            int v=e[i].to;
            if(cost[v]<=maxm&&dis[v]>dis[u]+e[i].dis)
            {
                dis[v]=dis[u]+e[i].dis;
                if(!vis[v])
                {
                    if(!q.empty()&&dis[v]<dis[q.front()] )
                    {
                        q.push_front(v);
                    }
                    else 
                    {
                        q.push_back(v);
                    }
                    vis[v]=1;
                }
            }
        }
    }
    if(dis[t]>flow)
    {
        return 0;
    }
    return 1;
}
main()
{
    cin>>n>>m>>s>>t>>flow;
    for(int i=1;i<=n;i++)
    {
        scanf("%lld",&cost[i]);
        maxn=max(maxn,cost[i]);
    }
    for(int i=1;i<=m;i++)
    {
        int x,y,z;
        scanf("%lld%lld%lld",&x,&y,&z);
        add(x,y,z);
        add(y,x,z);
    }
    int l=cost[s],r=maxn+1;
    while(l<r)
    {
        int mid=(l+r)>>1;
        if(spfa(mid))
        {
            r=mid;
        }
        else
        {
            l=mid+1;
        }
    }
    if(spfa(l))
    {
        cout<<l;
        return 0;
    }
    if(spfa(l-1))
    {
        cout<<l-1;
        return 0;
    }
    cout<<-1;
}

暂无评论

正在回复给  
去登陆?

   点击刷新验证码

标签云

文章留名