45分求助,感觉自己的思路和代码跟题解上讲的没区别

P1462 通往奥格瑞玛的道路

huangx607087 @ 2019-08-18 19:22:30

#include <bits/stdc++.h>
using namespace std;
int n,m,b,p[607087];
int e[607087],w[607087],nxt[607087],hd[607087];
int q[2607087],d[607087];
bool v[607087];
int spfa(int s,int t,int lim)
{
    if(p[s]>lim||p[t]>lim) return 0;
    memset(d,53,sizeof(d));
    memset(v,0,sizeof(v));
    d[s]=0,q[1]=s,v[s]=1;
    int f=1,r=1;
    while(f<=r)
    {
        int x=q[f++];
        int k=hd[x];
        while(k)
        {
            if(d[x]+w[k]<d[e[k]])
            {
                d[e[k]]=d[x]+w[k];
                if(!v[e[k]]&&p[e[k]]<=lim)
                {
                    q[++r]=e[k];
                    v[e[k]]=1;
                }
            }
            k=nxt[k];
        }
        v[x]=0;
    }
    if(d[t]>=b) return 0;
    else return 1;
}
int main()
{
    int i,j,k=0;
    scanf("%d%d%d",&n,&m,&b);
    for(i=1;i<=n;i++) scanf("%d",&p[i]);
    for(i=1;i<=n;i++)
    {
        int x,y,z;scanf("%d%d%d",&x,&y,&z);
        e[++k]=y,w[k]=z,nxt[k]=hd[x],hd[x]=k;
        e[++k]=x,w[k]=z,nxt[k]=hd[y],hd[y]=k;
    }
    if(!spfa(1,n,2001123000))
    {
        printf("AFK\n");
        return 0;
    }
    int l=max(p[1],p[n]),r=-1;
    for(i=1;i<=n;i++) r=max(r,p[i]);
    while(l<=r)
    {
        int mid=(l+r)/2;
        if(spfa(1,n,mid)) r=mid-1;
        else l=mid+1;
    }
    printf("%d\n",l);
    return 0;
}

|