最优贸易(反向建图+spfa）

由于这道题只让买一次，卖一次 ，那么这道题的本质就变为：

求从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;
}