最短路【洛谷P1462】 通往奥格瑞玛的道路

P1462 通往奥格瑞玛的道路

二分答案去决定每个点是否可达，这样就转化成一个变量，跑最短路即可。

单调性：答案越大越容易得到。

code：

#include <iostream>
#include <cstdio>
#include <queue>

using namespace std;

const int wx=500017;

inline int read(){
int sum=0,f=1; char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1; ch=getchar();}
while(ch>='0'&&ch<='9'){sum=(sum<<1)+(sum<<3)+ch-'0'; ch=getchar();}
return sum*f;
}

int num,n,m,hp,l,r,ans;
int dis[wx],vis[wx];
int head[wx],a[wx];

struct e{
int nxt,to,dis;
}edge[wx*2];

void add(int from,int to,int dis){
edge[++num].nxt=head[from];
edge[num].to=to;
edge[num].dis=dis;
head[from]=num;
}

struct node{
int u,d;
friend bool operator < (const node & a,const node & b){
return a.d>b.d;
}
};

priority_queue<node > q;

bool ok(int now){
for(int i=1;i<=n;i++)dis[i]=0x3f3f3f3f,vis[i]=0;
dis[1]=0; q.push((node){1,0});
while(q.size()){
int u=q.top().u; q.pop();
if(vis[u])continue; vis[u]=1;
for(int i=head[u];i;i=edge[i].nxt){
int v=edge[i].to; if(a[v]>now)continue;
if(dis[v]>dis[u]+edge[i].dis){
dis[v]=dis[u]+edge[i].dis;
q.push((node){v,dis[v]});
}
}
}
return dis
<=hp;
}

int main(){
n=read(); m=read(); hp=read();
for(int i=1;i<=n;i++)a[i]=read(),r=max(r,a[i]);
for(int i=1;i<=m;i++){
int x,y,z;
x=read(); y=read(); z=read();
add(x,y,z); add(y,x,z);
}
while(l<=r){
int mid=l+r>>1;
if(ok(mid))r=mid-1,ans=mid;
else l=mid+1;
}
if(!ans)puts("AFK");
else printf("%d\n",ans);
return 0;
}