洛谷 P1462 通往奥格瑞玛的道路

题目背景

在艾泽拉斯大陆上有一位名叫歪嘴哦的神奇术士，他是部落的中坚力量
有一天他醒来后发现自己居然到了联盟的主城暴风城
在被众多联盟的士兵攻击后，他决定逃回自己的家乡奥格瑞玛

题目描述

在艾泽拉斯，有n个城市。编号为1,2,3,…,n。
城市之间有m条双向的公路，连接着两个城市，从某个城市到另一个城市，会遭到联盟的攻击，进而损失一定的血量。
没经过一个城市，都会被收取一定的过路费（包括起点和终点）。路上并没有收费站。
假设1为暴风城，n为奥格瑞玛，而他的血量最多为b，出发时他的血量是满的。
歪嘴哦不希望花很多钱，他想知道，在可以到达奥格瑞玛的情况下，他所经过的所有城市中最多的一次收取的费用的最小值是多少。

输入输出格式

输入格式：

第一行3个正整数，n，m，b。分别表示有n个城市，m条公路，歪嘴哦的血量为b。
接下来有n行，每行1个正整数，fi。表示经过城市i，需要交费fi元。
再接下来有m行，每行3个正整数，ai，bi，ci(1<=ai，bi<=n)。表示城市ai和城市bi之间有一条公路，如果从城市ai到城市bi，或者从城市bi到城市ai，会损失ci的血量。

输出格式：

仅一个整数，表示歪嘴哦交费最多的一次的最小值。
如果他无法到达奥格瑞玛，输出AFK。

输入输出样例

输入样例#1：

4 4 8
8
5
6
10
2 1 2
2 4 1
1 3 4
3 4 3

输出样例#1：

10

说明

对于60%的数据，满足n≤200，m≤10000，b≤200
对于100%的数据，满足n≤10000，m≤50000，b≤1000000000
对于100%的数据，满足ci≤1000000000，fi≤1000000000，可能有两条边连接着相同的城市。

分析

直接二分最大费用是多少，然后跑一遍spfa，假如能够到达，就减少，不能就增多。

代码

#include <bits/stdc++.h>

#define maxn 20100
#define maxm 100100
#define ll long long

using namespace std;

ll n,m,b,Max=-1,cnt=1;

ll f[maxn],d[maxn],head[maxn];

struct edge
{
    ll u,next,c;
}e[maxm];
void adde(ll a,ll b,ll c)
{
    e[cnt].u=b;
    e[cnt].c=c;
    e[cnt].next=head[a];
    head[a]=cnt++;
}

ll vis[maxn];

bool ok =0;

bool spfa(ll mid)
{
    queue<ll> q;
    memset(vis,0,sizeof(vis));
    for(ll i=1;i<=n;i++)
        d[i]=1e15;
    d[1]=0;
    vis[1]=1;
    q.push(1);
    for(int i=1;i<=n;i++)
        if(f[i]>mid)
            vis[i]=1;
    if(f[1]>mid||f[n]>mid)
        return 0;
    while(!q.empty())
    {
        ll x=q.front();q.pop();
        vis[x]=0;
        for(ll i=head[x];i!=-1;i=e[i].next)
        {
            ll u=e[i].u;
            if(d[u]>d[x]+e[i].c)
            {
                d[u]=d[x]+e[i].c;
                if(!vis[u])
                {
                    vis[u]=1;
                    q.push(u);
                }
            }
        }
    }
    if(d[n]>b)
        return 0;
    ok=1;
    return 1;
}

int main()
{
    scanf("%lld%lld%lld",&n,&m,&b);
    memset(head,-1,sizeof(head));
    for(ll i=1;i<=n;i++)
    {
        scanf("%lld",&f[i]);
        Max=max(Max,f[i]);
    }
    for(ll i=1;i<=m;i++)
    {
        ll a,b,c;
        scanf("%lld%lld%lld",&a,&b,&c);
        adde(a,b,c);
        adde(b,a,c);
    }
    ll l=1,r=Max;
    ll mid;
    while(l<r)
    {
        mid=(l+r)/2;
        if(spfa(mid))
            r=mid;
        else l=mid+1;
    }
    if(!ok&&!spfa(l))
        printf("AFK");
        else printf("%lld",l);
    return 0;
}