【算法】最优贸易（反向建图）

题目

C 国有 n 个大城市和 m 条道路，每条道路连接这 n 个城市中的某两个城市。

任意两个城市之间最多只有一条道路直接相连。

这 m 条道路中有一部分为单向通行的道路，一部分为双向通行的道路，双向通行的道路在统计条数时也计为 1 条。

C 国幅员辽阔，各地的资源分布情况各不相同，这就导致了同一种商品在不同城市的价格不一定相同。

但是，同一种商品在同一个城市的买入价和卖出价始终是相同的。

商人阿龙来到 C 国旅游。

当他得知“同一种商品在不同城市的价格可能会不同”这一信息之后，便决定在旅游的同时，利用商品在不同城市中的差价赚一点旅费。

设 C 国 n 个城市的标号从 1∼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 ≤ n ≤ 1e5
1 ≤ m ≤ 5e5
1≤各城市水晶球价格≤100

输入样例：

5 5
4 3 5 6 1
1 2 1
1 4 1
2 3 2
3 5 1
4 5 2

输出样例：

5

思路 

  i 的取值范围：1 <= i <= n

如上图所示，从点 1 出发 经过点 i 最终到达点 n ,正向建图寻找点1到点 i 之间的价值最小的物品， 然后从点 n 出发寻找点 n 到点 i 的最大价值的物品。

然后如下图：

 从点 1 开始保留点最大值输出。

代码1

#include <bits/stdc++.h>
using namespace std;
typedef pair<int, int> PII;
const int INF = 0x3f3f3f3f;
const int N = 100010, M = 2000010;

int n, m;
int dist1[N], dist2[N];

// 正反建图，传入头数组指针
int h1[N], h2[N], e[M], ne[M], w[M], idx;
void add(int *h, int a, int b, int c = 0) {
    e[idx] = b, ne[idx] = h[a], w[idx] = c, h[a] = idx++;
}
// 每个节点的价值
int v[N];

void dijkstra1() {
    memset(dist1, 0x3f, sizeof dist1);
    priority_queue<PII, vector<PII>, greater<PII>> q;
    dist1[1] = v[1];
    q.push({dist1[1], 1});

    while (q.size()) {
        int u = q.top().second;
        q.pop();

        for (int i = h1[u]; ~i; i = ne[i]) {
            int j = e[i];
            if (dist1[j] > min(dist1[u], v[j])) {
                dist1[j] = min(dist1[u], v[j]);
                q.push({dist1[j], j});
            }
        }
    }
}

void dijkstra2() {
    memset(dist2, -0x3f, sizeof dist2);
    priority_queue<PII> q;
    dist2[n] = v[n];
    q.push({dist2[n], n});

    while (q.size()) {
        int u = q.top().second;
        q.pop();
        for (int i = h2[u]; ~i; i = ne[i]) {
            int j = e[i];
            if (dist2[j] < max(dist2[u], v[j])) {
                dist2[j] = max(dist2[u], v[j]);
                q.push({dist2[j], j});
            }
        }
    }
}

int main() {
    // 正反两张图
    //  Q:为什么要反着建图，用正着的图不行吗？
    //  A:不行啊，因为从n向其它地方走，原来的有向图无法向对面走啊，反着建图就行了
    memset(h1, -1, sizeof h1);
    memset(h2, -1, sizeof h2);

    scanf("%d %d", &n, &m);                          // n个节点，m条边
    for (int i = 1; i <= n; i++) scanf("%d", &v[i]); // 每个节点购买水晶球的金额

    while (m--) {
        int a, b, c;
        scanf("%d %d %d", &a, &b, &c);
        // 不管是单向边，还是双向边，第一条a->b的边肯定跑不了吧

        if (c == 1) { // 单向边
            // 正向图保存单向边
            add(h1, a, b);
            // 反向图保存单向边
            add(h2, b, a);
            // 注意：这可不是在一个图中创建两条来回的边，而是在两个图中创建两个相反的边。
            // 权值呢？没有，为什么呢？因为我们不关心边权，而是关心此节点中水晶球的价格v[i]，这并不是边权,可以理解为点权
        } else { // 双向边
            // 正向图保存双向边
            add(h1, a, b), add(h1, b, a);
            // 反向图保存双向边
            add(h2, a, b), add(h2, b, a);
        }
    }
    // 正向图跑一遍dijkstra
    dijkstra1();
    // 反向图跑一遍dijkstra
    dijkstra2();

    int ans = 0;
    for (int i = 1; i <= n; i++)
        ans = max(dist2[i] - dist1[i], ans);

    printf("%d\n", ans);
    return 0;
}

代码2

#include<bits/stdc++.h>
using namespace std;
typedef pair<int,int> PII;
const int N = 1e5 + 10,M = 5e5 + 10;
int n,m;
int w[N];
int h1[N],ne1[M],e1[M],idx1;
int h2[N],ne2[M],e2[M],idx2;
int dist_min[N];
int dist_max[N];

void add1(int a,int b)
{
    ne1[idx1] = h1[a],e1[idx1] = b,h1[a] = idx1 ++;
}

void add2(int a,int b)
{
    ne2[idx2] = h2[a],e2[idx2] = b,h2[a] = idx2 ++;
}

void dijkstra1()
{
    priority_queue<PII,vector<PII>,greater<>> heap;
    memset(dist_min,0x3f,sizeof dist_min);
    dist_min[1] = w[1];
    heap.emplace(w[1],1);
    while(!heap.empty())
    {
        auto t = heap.top();
        heap.pop();
        int x = t.second;
        for(int i = h1[x]; ~i; i = ne1[i])
        {
            int j = e1[i];
            if(dist_min[j] > min(dist_min[x],w[j]))
            {
                dist_min[j] = min(dist_min[x],w[j]);
                heap.emplace(dist_min[j],j);
            }
        }
    }
}

void dijkstra2()
{
    priority_queue<PII,vector<PII>,less<>> heap;
    memset(dist_max,0,sizeof dist_max);
    dist_max[n] = w[n];
    heap.emplace(w[n],n);
    while(!heap.empty())
    {
        auto t = heap.top();
        heap.pop();
        int x = t.second;
        for(int i = h2[x]; ~i; i = ne2[i])
        {
            int j = e2[i];
            if(dist_max[j] < max(dist_max[x],w[j]))
            {
                dist_max[j] = max(dist_max[x],w[j]);
                heap.emplace(dist_max[j],j);
            }
            
        }
    }
}

int main()
{
    cin >> n >> m;
    memset(h1,-1,sizeof h1);
    memset(h2,-1,sizeof h2);
    for(int i = 1; i <= n; i ++) cin >> w[i];
    while(m --)
    {
        int op,a,b;
        cin >> a >> b >> op;
        if(op == 1)
        {
            add1(a,b);
            add2(b,a);
        }
        else
        {
            add1(a,b),add1(b,a);
            add2(a,b),add2(b,a);
        }
    }
    dijkstra1();
    dijkstra2();
    int ans = 0;
    for(int i = 1; i <= n; i ++) ans = max(ans,dist_max[i] - dist_min[i]);
    cout << ans << endl;
    return 0;
}

难度：中等

时/空限制：1s / 64MB

总通过数：9946

总尝试数：19663

来源：《算法竞赛进阶指南》, NOIP2009提高组

算法标签 图论  SPFA

题目来自：341. 最优贸易 - AcWing题库