Luogu P1073 最优贸易

题目

P1073 最优贸易

分析

分不清这道题到底是DP还是SPFA……
显然这道题是要使得买入价格最低，卖出价格最高。跟数列不同的是，图是有访问顺序的。不妨从起点到终点方向找一个最小买入点，从终点往起点反向找一个最大卖出点，两者作差就是最大收益了。这样的好处是不需考虑买入点是否能走到卖出点，因为反向建图找卖出点，就意味着有起点方向的边（如果不理解可以慢慢往下看，下一段有更形象的解释）。那么将条边的边权设为它的终点点权，初始化每个点最小买入价值为极大值，每次取尽量小的价值，一直到终点；反向建图，初始化最小卖出价值为极小值，每次取尽量大的价值一直到起点。
这样dis[i]表示从起点到i以来，最小的买入价值；nes[i]表示，从终点反向到i以来，最大的卖出价值。由于反向边能从终点到i，意味着正向边可以从i到终点，而dis[i]保存的是i及i之前的某一点的最小值，nes[i]保存的是i及i之后的某一点的最大值，所以dis[i]存储的这个点，一定能到达nes[i]存储的点。所以保证了nes[i]-dis[i]是一个可能的价值。
最后取各个点中最大的nes[i]-dis[i]，即为答案。
借用dp的思想，SPFA更新时取前一条边跟当前边比较的最小值，作为从起点到此点时，最后小的买入价值。反向类似。

代码

#include<cstdio>
#include<queue>
#include<cstring>
#include<iostream>
using namespace std;
const int maxn=100002,maxm=500002,inf=2147483647;
struct Edge{
    int to,next,v;
}e[maxm*2],g[maxm*2];
int heade[maxn],headg[maxn],a[maxn],dis[maxn],nes[maxn];
bool vis[maxn];
queue<int>q;
int n,m,cnt,num,ans;
void adde(int u,int v,int w)//正向图加边
{
    e[++cnt].to=v;
    e[cnt].next=heade[u];
    e[cnt].v=w;
    heade[u]=cnt;
}
void addg(int u,int v,int w)//反向图加边
{
    g[++num].to=v;
    g[num].next=headg[u];
    g[num].v=w;
    headg[u]=num;
}
void spfae()
{
    memset(dis,127,sizeof(dis));
    dis[1]=a[1];
    q.push(1);
    vis[1]=1;
    while(!q.empty())
    {
        int u=q.front();
        q.pop();
        vis[u]=0;
        for(int i=heade[u];i;i=e[i].next)
        {
            int v=e[i].to;
            if(dis[v]>min(dis[u],e[i].v))
            {
                dis[v]=min(dis[u],e[i].v);
                if(!vis[v])
                {
                    vis[v]=1;
                    q.push(v);
                }
            }
        }
    }
}
void spfag()
{
    memset(vis,0,sizeof(vis));
    memset(nes,128,sizeof(nes));
    nes[n]=a[n];
    q.push(n);
    vis[n]=0;
    while(!q.empty())
    {
        int u=q.front();
        q.pop();
        vis[u]=0;
        for(int i=headg[u];i;i=g[i].next)
        {
            int v=g[i].to;
            if(nes[v]<max(nes[u],g[i].v))
            {
                nes[v]=max(nes[u],g[i].v);
                if(!vis[v])
                {
                    vis[v]=1;
                    q.push(v);  
                }
            }
        }
    }
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
        scanf("%d",&a[i]);
    for(int i=1,x,y,z;i<=m;i++)
    {
        scanf("%d%d%d",&x,&y,&z);
        if(z==1)    adde(x,y,a[y]),addg(y,x,a[x]);
        else        adde(x,y,a[y]),adde(y,x,a[x]),addg(y,x,a[y]),addg(x,y,a[x]);
    }
    spfae();
    spfag();
    for(int i=1;i<=n;i++)
        ans=max(ans,nes[i]-dis[i]);
    printf("%d",ans);
    return 0;
}