由于这道题只让买一次,卖一次 ,那么这道题的本质就变为:
求从1~i最小的花费,和从i~n最大卖出去的钱,最后枚举每个i,取max(dmax[i]-dmin[i])
本题还有一个重点就是,从i~n,可以把图反向建边,然后求n~i的值,这样就只就行一次spfa
代码
#include<iostream>
#include<algorithm>
#include<cstring>
#include<queue>
using namespace std;
const int N = 100010, M = 2000010;
int ht[N],he[N],e[M],ne[M],idx;
int w[N];
int dmax[N],dmin[N];
int n,m;
bool st[N];
void add(int h[],int a,int b)
{
e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}
void spfa(int h[],int type)
{
memset(st,false,sizeof st);
queue<int> q;
if(type==0)
{
memset(dmin,0x3f,sizeof dmin);
dmin[1]=w[1];
st[1]=true;
q.push(1);
}
else
{
memset(dmax,-0x3f,sizeof dmax);
dmax[n]=w[n];
st[n]=true;
q.push(n);
}
while(q.size())
{
auto t=q.front();
q.pop();
st[t]=false;
for(int i=h[t];~i;i=ne[i])
{
int j=e[i];
if( ( type==0 && dmin[j]>min(dmin[t],w[j]) ) || (type==1 && dmax[j]<max(dmax[t],w[j])) )
{
if(type==0) dmin[j]=min(dmin[t],w[j]);
else dmax[j]=max(dmax[t],w[j]);
if(!st[j])
{
st[j]=true;
q.push(j);
}
}
}
}
}
int main()
{
cin>>n>>m;
memset(ht,-1,sizeof ht);
memset(he,-1,sizeof he);
for(int i=1;i<=n;i++) scanf("%d",&w[i]);
while(m--)
{
int a,b,c;
scanf("%d%d%d",&a,&b,&c);
add(ht,a,b);add(he,b,a);
if(c==2) add(ht,b,a),add(he,a,b);
}
spfa(ht,0);
spfa(he,1);
int res=-1;
for(int i=1;i<=n;i++) res=max(res,dmax[i]-dmin[i]);
cout<<res<<endl;
return 0;
}