牛客 - 最短路 - [NOIP2009]最优贸易

溏心荷包蛋798

于 2024-05-06 22:08:27 发布

阅读量1.2k

点赞数 37

分类专栏：牛客文章标签： c++ 算法

本文链接：https://blog.csdn.net/Hahahaha5346/article/details/138510107

版权

牛客专栏收录该内容

8 篇文章 0 订阅

订阅专栏

牛客 - 最短路 - [NOIP2009]最优贸易

链接：https://ac.nowcoder.com/acm/problem/16611
来源：牛客网

题目描述

C国有n个大城市和m条道路，每条道路连接这n个城市中的某两个城市。任意两个城市之间最多只有一条道路直接相连。这m条道路中有一部分为单向通行的道路，一部分为双向通行的道路，双向通行的道路在统计条数时也计为1条。

C国幅员辽阔，各地的资源分布情况各不相同，这就导致了同一种商品在不同城市的价格不一定相同。但是，同一种商品在同一个城市的买入价和卖出价始终是相同的。

商人阿龙来到C国旅游。当他得知同一种商品在不同城市的价格可能会不同这一信息之后，便决定在旅游的同时，利用商品在不同城市中的差价赚回一点旅费。设C国n个城市的标号从1-n，阿龙决定从1号城市出发，并最终在n号城市结束自己的旅行。在旅游的过程中，任何城市可以重复经过多次，但不要求经过所有n个城市。阿龙通过这样的贸易方式赚取旅费：他会选择一个经过的城市买入他最喜欢的商品——水晶球，并在之后经过的另一个城市卖出这个水晶球。用赚取的差价当作旅费。由于阿龙主要是来C国旅游，他决定这个贸易只进行最多一次。当然，在赚不到差价的情况下它就无需进行贸易。

假设C国有5个大城市，城市的编号和道路连接情况如下图，单向箭头表示这条道路为单向通行。双向箭头表示这条道路为双向通行。

假设1~n号城市的水晶球价格分别为4，3，5，6，1。

阿龙可以选择如下一条线路：1->2->3->5，并在2号城市以3的价格买入水晶球，在3号城市以5的价格卖出水晶球，赚取的旅费数为2。

阿龙也可以选择如下一条线路：1->4->5->4->5，并在第1次到达5号城市时以1的价格买入水晶球，在第2次到达4号城市时以6的价格卖出水晶球，赚取的旅费数为5。

现在给出n个城市的水晶球价格，m条道路的信息（每条道路所连接的两个城市的编号以及该条道路的通行情况）。请你告诉阿龙，他最多能赚钱多少旅费。

输入描述:

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

输出描述:

共1行，包含1个整数，表示最多能赚取的旅费。如果没有进行贸易，则输出0。

示例1

输入

输出

思路分析：

算法
(SPFA) O(n+km)
先求出：从 1走到 i的过程中，买入水晶球的最低价格 dmin[i]；从 i走到 n的过程中，卖出水晶球的最高价格 dmax[i]；
然后枚举每个城市作为买卖的中间城市，求出 dmax[i] - dmin[i] 的最大值即可。

求 dmin[i] 和 dmax[i] 时，由于不是拓扑图，状态的更新可能存在环，因此不能使用动态规划，只能使用求最短路的方式。
另外，我们考虑能否使用 dijkstra 算法，如果当前 dmin[i] 最小的点是 5，那么有可能存在边 5-> 6, 6-> 7, 7-> 5，假设当前 dmin[5] = 10，则有可能存在 6 的价格是11，但 7 的价格是3，那么 dmin[5] 的值就应该被更新成3，因此当前最小值也不一定是最终最小值，所以dijkstra算法并不适用，我们只能采用 spfa 算法。

时间复杂度瓶颈是SPFA，SPFA 算法的时间复杂度是 O(km)
其中 k一般情况下是个很小的常数，最坏情况下是 n 表示总点数，m表示总边数。因此总时间复杂度是 O(km)

#include<bits/stdc++.h>

using namespace std;

const int N = 100010, M = 500010;
int h[N], rh[N], e[M], ne[M], idx;
int n, m;
int price[N]; //存水晶球的价格
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 ++;
}

void spfa(int *d, int start, int *h, bool flag) //基本都是spfa的模板
{
    queue<int> q;
    memset(st, 0, sizeof st);
    
    if(flag) memset(dmin, 0x3f, sizeof dmin); //dmin需要初始化成无穷大
    
    q.push(start);
    st[start] = true;
    d[start] = price[start];
    
    while(!q.empty())
    {
        int t = q.front();
        q.pop();
        st[t] = false;
        
        for(int i=h[t]; i!=-1; i=ne[i])
        {
            int j = e[i];//取出头节点
            if(flag && d[j]>min(d[t], price[j]) || !flag && d[j]<max(d[t], price[j]))//两种情况
            {
                //spfa 中的三角不等式
                if(flag) d[j] = min(d[t], price[j]);
                else d[j] = max(d[t], price[j]);
                
                if(!st[j]) //队列中不放重复的点，需要判断一下
                {
                    st[j] = true; 
                    q.push(j);
                }
            }
        }
    }
}

int main()
{
    memset(h, -1, sizeof h);//初始化头结点
    memset(rh, -1, sizeof rh);
    
    cin >> n >> m;
    
    for(int i=1; i<=n; i++) cin >> price[i];
    
    while(m --)
    {
        int a, b ,z;
        cin >> a >> b >> z;
        add(h, a, b);
        add(rh, b, a);
        if(z == 2)
        {
            add(h, b, a);
            add(rh, a, b);
        }   
    }
    
    spfa(dmin, 1, h, true);
    spfa(dmax, n, rh, false);
    
    int res = 0;
    for(int i=1; i<=n; i++) res = max(res, dmax[i] - dmin[i]); //最后遍历一下答案
    
    cout << res;
    
    return 0;
}