【ACWing】341. 最优贸易

题目地址：

https://www.acwing.com/problem/content/343/

$C$国有$n$个大城市和$m$条道路，每条道路连接这$n$个城市中的某两个城市。任意两个城市之间最多只有一条道路直接相连。这$m$条道路中有一部分为单向通行的道路，一部分为双向通行的道路，双向通行的道路在统计条数时也计为$1$条。$C$国幅员辽阔，各地的资源分布情况各不相同，这就导致了同一种商品在不同城市的价格不一定相同。但是，同一种商品在同一个城市的买入价和卖出价始终是相同的。商人阿龙来到$C$国旅游。当他得知“同一种商品在不同城市的价格可能会不同”这一信息之后，便决定在旅游的同时，利用商品在不同城市中的差价赚一点旅费。设$C$国$n$个城市的标号从$1\sim n$，阿龙决定从$1$号城市出发，并最终在$n$号城市结束自己的旅行。在旅游的过程中，任何城市可以被重复经过多次，但不要求经过所有$n$个城市。阿龙通过这样的贸易方式赚取旅费：他会选择一个经过的城市买入他最喜欢的商品——水晶球，并在之后经过的另一个城市卖出这个水晶球，用赚取的差价当做旅费。因为阿龙主要是来$C$国旅游，他决定这个贸易只进行最多一次，当然，在赚不到差价的情况下他就无需进行贸易。现在给出$n$个城市的水晶球价格，$m$条道路的信息（每条道路所连接的两个城市的编号以及该条道路的通行情况）。请你告诉阿龙，他最多能赚取多少旅费。注意：本题数据有加强。

输入格式：
第一行包含$2$个正整数$n$和$m$，中间用一个空格隔开，分别表示城市的数目和道路的数目。第二行$n$个正整数，每两个整数之间用一个空格隔开，按标号顺序分别表示这$n$个城市的商品价格。接下来$m$行，每行有$3$个正整数，$x$，$y$，$z$，每两个整数之间用一个空格隔开。如果$z=1$，表示这条道路是城市$x$到城市$y$之间的单向道路；如果$z=2$，表示这条道路为城市$x$和城市$y$之间的双向道路。

输出格式：
一个整数，表示答案。

数据范围：
$1\le n\le 100000$
$1\le m\le 500000$
$1\le price\le 100$

每条路径可以按照买卖之间经过的点来分类，设路径是从$1$先走到$i$，再走到$n$的，并且买在走到$i$之前，卖在走到$i$之后，那么最大差价就是从$1$走到$i$的价格最小值（包括点$i$，这是买点），减去从$i$走到$n$的价格最大值（包括点$i$，这是卖点），再枚举所有的$i$就得到了全局答案。设从$1$走到$i$的最小值是$f[i]$，设从$i$走到$n$的最大值是$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$同时包含在$f$和$g$的计算里，是因为在同一点买卖的话差价是$0$，相当于不做交易，而不做交易是允许的，并且写起来也更方便（否则的话要枚举$i$能走到的点$j$，然后算$g[j]-f[i]$，这样写只会增加麻烦）。所以问题就转化为求$f[i]$和$g[i]$。设能走到$i$的点分别是$x_1,...,x_j$，$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_j$点或其之前买，或者在点$i$处买。但是我们不能直接按上式递推地来求，因为有可能图里有环，不存在拓扑序。这里要用类似Bellman-Ford最短路的方法来求，即松弛$n-1$次的方法。设$a[k][i]$是走不超过$k$条边的情况下，从$1$走到$i$的最小值，初始条件$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$走到$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(nm)$（这是SPFA的最差时间复杂度，真实时间不会这么多），空间$O(n)$。