我们在做这道题时正反图要同时建立,然后从起点出发,跑一遍DJ处理出权值最小的点F[x],再反着跑一遍处理出权值最大的点,D[x]。
再枚举每个点,用D[x]-F[x]跟新答案即可。
#include<bits/stdc++.h>
#define maxn 510000
using namespace std;
priority_queue< pair < int , int > >qs;
priority_queue< pair < int , int > >qf;
struct edge_shun
{
int to,pre;
}es[maxn];
struct edge_fan
{
int to,pre;
}ef[maxn];
int n,m,a[maxn],cnts=0,cntf=0,heads[maxn],headf[maxn],D[maxn],F[maxn];
bool used[maxn];
void add_edge(int x,int y,int z)
{
es[++cnts].to=y;
es[cnts].pre=heads[x];
heads[x]=cnts;
ef[++cntf].to=x;
ef[cntf].pre=headf[y];
headf[y]=cntf;
}
void read(int &x)
{
int f=1;x=0;
char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
x*=f;
}
void mem()
{
for(int i=1;i<=n;i++)
used[i]=true;
}
void ds()
{
mem();
D[1]=a[1];
qs.push(make_pair(-D[1],1));
while(qs.size())
{
int x=qs.top().second;
qs.pop();
if(used[x]==false) continue;
used[x]=false;
for(int i=heads[x];i;i=es[i].pre)
{
int v=es[i].to;
if(min(D[x],a[v])<D[v])
{
D[v]=min(D[x],a[v]);
qs.push(make_pair(-D[v],v));
}
}
}
}
void df()
{
mem();
F[n]=a[n];
qf.push(make_pair(F[n],n));
while(qf.size())
{
int x=qf.top().second;
qf.pop();
if(used[x]==false) continue;
used[x]=false;
for(int i=headf[x];i;i=ef[i].pre)
{
int v=ef[i].to;
if(max(F[x],a[v])>F[v])
{
F[v]=max(F[x],a[v]);
qf.push(make_pair(F[v],v));
}
}
}
}
int main()
{
read(n);read(m);
for(int i=1;i<=n;i++)
read(a[i]);
for(int i=1;i<=m;i++)
{
int x,y,z;
read(x);read(y);read(z);
if(z==1)
add_edge(x,y,z);
else
{
add_edge(x,y,z);
add_edge(y,x,z);
}
}
for(int i=1;i<=n;i++)
used[i]=true,D[i]=0x7ffffff,F[i]=-1;
ds();
df();
int ans=-1;
for(int i=1;i<=n;i++)
ans=max(ans,F[i]-D[i]);
printf("%d",ans);
return 0;
}