题目地址:
https://www.acwing.com/problem/content/343/
C C C国有 n n n个大城市和 m m m条道路,每条道路连接这 n n n个城市中的某两个城市。任意两个城市之间最多只有一条道路直接相连。这 m m m条道路中有一部分为单向通行的道路,一部分为双向通行的道路,双向通行的道路在统计条数时也计为 1 1 1条。 C C C国幅员辽阔,各地的资源分布情况各不相同,这就导致了同一种商品在不同城市的价格不一定相同。但是,同一种商品在同一个城市的买入价和卖出价始终是相同的。商人阿龙来到 C C C国旅游。当他得知“同一种商品在不同城市的价格可能会不同”这一信息之后,便决定在旅游的同时,利用商品在不同城市中的差价赚一点旅费。设 C C C国 n n n个城市的标号从 1 ∼ n 1∼n 1∼n,阿龙决定从 1 1 1号城市出发,并最终在 n n n号城市结束自己的旅行。在旅游的过程中,任何城市可以被重复经过多次,但不要求经过所有 n n n个城市。阿龙通过这样的贸易方式赚取旅费:他会选择一个经过的城市买入他最喜欢的商品——水晶球,并在之后经过的另一个城市卖出这个水晶球,用赚取的差价当做旅费。因为阿龙主要是来 C C C国旅游,他决定这个贸易只进行最多一次,当然,在赚不到差价的情况下他就无需进行贸易。现在给出 n n n个城市的水晶球价格, m m m条道路的信息(每条道路所连接的两个城市的编号以及该条道路的通行情况)。请你告诉阿龙,他最多能赚取多少旅费。注意:本题数据有加强。
输入格式:
第一行包含
2
2
2个正整数
n
n
n和
m
m
m,中间用一个空格隔开,分别表示城市的数目和道路的数目。第二行
n
n
n个正整数,每两个整数之间用一个空格隔开,按标号顺序分别表示这
n
n
n个城市的商品价格。接下来
m
m
m行,每行有
3
3
3个正整数,
x
x
x,
y
y
y,
z
z
z,每两个整数之间用一个空格隔开。如果
z
=
1
z=1
z=1,表示这条道路是城市
x
x
x到城市
y
y
y之间的单向道路;如果
z
=
2
z=2
z=2,表示这条道路为城市
x
x
x和城市
y
y
y之间的双向道路。
输出格式:
一个整数,表示答案。
数据范围:
1
≤
n
≤
100000
1≤n≤100000
1≤n≤100000
1
≤
m
≤
500000
1≤m≤500000
1≤m≤500000
1
≤
p
r
i
c
e
≤
100
1≤price≤100
1≤price≤100
每条路径可以按照买卖之间经过的点来分类,设路径是从 1 1 1先走到 i i i,再走到 n n n的,并且买在走到 i i i之前,卖在走到 i i i之后,那么最大差价就是从 1 1 1走到 i i i的价格最小值(包括点 i i i,这是买点),减去从 i i i走到 n n n的价格最大值(包括点 i i i,这是卖点),再枚举所有的 i i i就得到了全局答案。设从 1 1 1走到 i i i的最小值是 f [ i ] f[i] f[i],设从 i i i走到 n n n的最大值是 g [ i ] g[i] g[i],那么解就是 max 1 ≤ i ≤ n { g [ i ] − f [ i ] } \max_{1\le i\le n}\{g[i]-f[i]\} max1≤i≤n{g[i]−f[i]}。这里之所以要将 i i i同时包含在 f f f和 g g g的计算里,是因为在同一点买卖的话差价是 0 0 0,相当于不做交易,而不做交易是允许的,并且写起来也更方便(否则的话要枚举 i i i能走到的点 j j j,然后算 g [ j ] − f [ i ] g[j]-f[i] g[j]−f[i],这样写只会增加麻烦)。所以问题就转化为求 f [ i ] f[i] f[i]和 g [ i ] g[i] g[i]。设能走到 i i i的点分别是 x 1 , . . . , x j x_1,...,x_j x1,...,xj, i i i号点的商品价格是 c [ i ] c[i] c[i],那么就有: f [ i ] = min { f [ x 1 ] , f [ x 2 ] , . . . , f [ x j ] , c [ i ] } f[i]=\min \{f[x_1],f[x_2],...,f[x_j],c[i]\} f[i]=min{f[x1],f[x2],...,f[xj],c[i]}分别对应着在 x 1 x_1 x1点或其之前买,一直到在 x j x_j xj点或其之前买,或者在点 i i i处买。但是我们不能直接按上式递推地来求,因为有可能图里有环,不存在拓扑序。这里要用类似Bellman-Ford最短路的方法来求,即松弛 n − 1 n-1 n−1次的方法。设 a [ k ] [ i ] a[k][i] a[k][i]是走不超过 k k k条边的情况下,从 1 1 1走到 i i i的最小值,初始条件 a [ 0 ] [ 1 ] = c [ 1 ] a[0][1]=c[1] a[0][1]=c[1],递推方程为: a [ k ] [ i ] = min { a [ k − 1 ] [ x 1 ] , a [ k − 1 ] [ x 2 ] , . . . , a [ k − 1 ] [ x j ] , c [ i ] } a[k][i]=\min\{a[k-1][x_1],a[k-1][x_2],...,a[k-1][x_j],c[i]\} a[k][i]=min{a[k−1][x1],a[k−1][x2],...,a[k−1][xj],c[i]}而Bellman-Ford算法可以用队列优化,即变为SPFA算法,而不影响正确性。至于从 i i i走到 n n n的最大值,可以建反向边,然后将 n n n做起点做一遍SPFA即可。代码如下:
#include <iostream>
#include <cstring>
#include <queue>
using namespace std;
// 边要开4倍,因为每个边可能是双向边,然后还要建反向图
const int N = 100010, M = 2000010;
int n, m;
// w[i]表示城市i的物品价格
int w[N];
// hs存正向图,ht存反向图
int hs[N], ht[N], e[M], ne[M], idx;
// dmin[i]表示从点1到点i的路径里的物品最小价格,dmax[i]表示从点n到点i的沿反向图走的路径里的物品最大价格
int dmin[N], dmax[N];
bool st[N];
void add(int h[], int a, int b) {
e[idx] = b, ne[idx] = h[a], h[a] = idx++;
}
// h代表是在正向图还是反向图上考虑问题,type代表是算最小值还是最大值,0代表最小值,1代表最大值
void spfa(int h[], int dist[], int type) {
queue<int> q;
if (type == 0) {
// 这里的sizeof后面一定要写dmin,dist这里只是个指针而不是数组
memset(dist, 0x3f, sizeof dmin);
dist[1] = w[1];
q.push(1);
} else {
dist[n] = w[n];
q.push(n);
}
while (q.size()) {
int t = q.front();
q.pop();
st[t] = false;
for (int i = h[t]; ~i; i = ne[i]) {
int j = e[i];
if (type == 0 && dist[j] > min(dist[t], w[j]) || type == 1 && dist[j] < max(dist[t], w[j])) {
if (type == 0) dist[j] = min(dist[t], w[j]);
else dist[j] = max(dist[t], w[j]);
if (!st[j]) {
q.push(j);
st[j] = true;
}
}
}
}
}
int main() {
scanf("%d%d", &n, &m);
for (int i = 1; i <= n; i++) scanf("%d", &w[i]);
memset(hs, -1, sizeof hs);
memset(ht, -1, sizeof ht);
while (m--) {
int a, b, c;
scanf("%d%d%d", &a, &b, &c);
// 建从a到b的边和其反向边
add(hs, a, b), add(ht, b, a);
// 如果边是双向的,还要建从b到a的边和其反向边
if (c == 2) add(hs, b, a), add(ht, a, b);
}
// 分别求一下从1出发到每个顶点的路径最小价格,和从n出发按反向图走的到每个顶点的路径最大价格
spfa(hs, dmin, 0);
spfa(ht, dmax, 1);
int res = 0;
for (int i = 1; i <= n; i++)
res = max(res, dmax[i] - dmin[i]);
cout << res << endl;
return 0;
}
时间复杂度 O ( n m ) O(nm) O(nm)(这是SPFA的最差时间复杂度,真实时间不会这么多),空间 O ( n ) O(n) O(n)。