题目
分析
分不清这道题到底是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;
}