AcWing 341. 最优贸易

题目链接

题目思路

题目的要求是找出一条路径中的最大点权减去最小点权值最大
我们可以spfa正向求一遍最小路 再反向求一遍最大路
对于每个点的最小值dmin即为从起点到此点所经历的最小点权值
对于每个点的最大值dmax即为从终点到此点所经历的最大点权值
有一个注意的事项
在函数部分中 如果你没设一个全局变量d而是函数变量 memset的时候要sizeof你所选的那个全局变量

---

代码

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
#include <queue>

using namespace std;

const int N = 100010, M = 1000010;

int n, m;
int h[N], rh[N], e[M], ne[M], idx;
int w[N], 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 ++ ;
}

void spfa(int d[], int u, int h[], bool flag)
{
    queue<int> q;
    memset(st, 0, sizeof st);
    
    if (flag) memset(d, 0x3f, sizeof dmin);
    
    q.push(u), d[u] = w[u], st[u] = true;;
    
    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 (flag && d[j] > min(d[t], w[j]) || !flag && d[j] < max(d[t], w[j]))
            {
                if (flag) d[j] = min(d[t], w[j]);
                else d[j] = max(d[t], w[j]);
                
                if (!st[j])
                {
                    st[j] = true;
                    q.push(j);
                }
            }
        }
    }
}

int main()
{
    scanf("%d%d", &n, &m);
    
    for (int i = 1; i <= n; i ++ ) scanf("%d", &w[i]);
    
    memset(h, -1, sizeof h);
    memset(rh, -1, sizeof rh);
    
    while (m -- )
    {
        int a, b, c;
        scanf("%d%d%d", &a, &b, &c);
        add(h, a, b), add(rh, b, a);
        if (c == 2)
            add(h, b, a), add(rh, a, b);
    }
    
    spfa(dmin, 1, h, true);
    spfa(dmax, n, rh, false);
    
    int ans = 0;
    for (int i = 1; i <= n; i ++ ) ans = max(ans, dmax[i] - dmin[i]);
    
    printf("%d\n", ans);
    
    return 0;
}