[luogu1462]通往奥格瑞玛的道路

题目链接

题解：二分一下

#include <iostream>
#include <cstring>
#include <cstdio>
#include <queue>
using namespace std;
#define INF 0x3f3f3f3f
const int M=10005;
int m,n,x,y,z,t,w,r,l,mid,ans;
int head[M],d[M],p[M];
bool vis[M],mark[M];
queue <int> q;

struct edge{
    int from,to,next,val;
}e[M*10];

void ready(int s)
{
    for(int i=1;i<=n;i++)
    d[i]=INF,vis[i]=0;
    d[s]=0;
}

void add(int i,int j,int v)
{
    e[t].from=i;
    e[t].to=j;
    e[t].val=v;
    e[t].next=head[i];
    head[i]=t++;
}

inline void spfa(int s)
{
    ready(s);
    q.push(s);
    while(!q.empty())
    {
        int u=q.front();q.pop();vis[u]=0;
        for(int i=head[u];i!=-1;i=e[i].next)
        {
            int v=e[i].to;
            if(mark[v]&&d[v]>d[u]+e[i].val)
            {
                d[v]=d[u]+e[i].val;
                if(!vis[v])
                {
                    q.push(v);
                    vis[v]=1;
                }
            }
        }
    }
}
bool ok(int x)
{
    for(int i=1;i<=n;i++)
    {
        if(p[i]<=x) mark[i]=true;//小于二分值的点可以走
        else mark[i]=false;
    }
    spfa(1);
    return d[n]<=w;//血量要小于w
}
void work()
{
    while(l<=r)
    {
        mid=(l+r)>>1;
        if(ok(mid)) r=mid-1,ans=mid;
        else l=mid+1;
    }
    if(!ans)
    printf("AFK\n");
    else
    printf("%d\n",ans);
}
void init()
{
    memset(head,-1,sizeof(head));
    cin>>n>>m>>w;
    for(int i=1;i<=n;i++)
    scanf("%d",&p[i]),r=max(r,p[i]);//二分max
    while(m--)
    {
        scanf("%d%d%d",&x,&y,&z);
        add(x,y,z);
        add(y,x,z);
    }
}
int main()
{
    init();
    work();
    return 0;
}