题目链接:
题目大意:
有N个城镇,M条路,路只能从编号小的城镇到编号大的城镇。每个城镇都有买卖黄金的贸易,不同城镇的黄金价格不同(但同一城镇买入和卖出价格相同),你按照路的顺序走,买卖一次黄金,问最多能赚多少钱?
思路:
这题和leetcode 121 买卖股票的最佳时机很像,只不过换成了图问题。买卖股票那题是可以用贪心来做的,所以想到了贪心思路,从每个入度为0的节点开始,遍历每一条路,计算最大利润。但是这样做会TLE。
原因是:当不同的入度为0的节点路线汇合的时候,这种算法会有大量的重复计算。所以要改成用dp解决,在计算汇合部分之前,先算出汇合点处的最低买入价格(该点的dp值)。计算完一个入度为0的节点之后,这个节点就可以被忽略,则视为被它指向的点入度-1,如此不断循环地把入度为1的节点加入计算,就可以保证每个结点只计算一遍,时间复杂度O(N)。
代码:
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <vector>
#include <iomanip>
#include <string.h>
#include <climits>
#include <queue>
typedef long long ll;
using namespace std;
const int maxn = 2e5+5;
const int INF = 0x3f3f3f3f;
vector<int> g[maxn]; // 记录连接
int val[maxn]; // 本地黄金价格
int dp[maxn]; // 记录本条路上最小的买入黄金价格
int in[maxn]; // 记录每个点的入度,用于拓扑排序
queue<int> q; // 按拓扑排序顺序处理节点
int main(){
int n, m;
cin >> n >> m;
for(int i=1; i<=n; i++){
cin >> val[i];
dp[i] = val[i]; //第一次初始化是本地价格
}
for(int i=1; i<=m; i++){
int a, b;
cin >> a >> b;
g[a].push_back(b);
in[b]++;
}
for(int i=1; i<=n; i++){
if(in[i] == 0) q.push(i); // 加入入度为0的节点
}
int ans = -INF; //ans可能为负数,所以要初始化为-INF
while(q.size()){
int u = q.front(); q.pop();
for(auto v : g[u]){
if(ans == -INF || ans < val[v]-dp[u]) ans = val[v] - dp[u]; // 更新最大利润
dp[v] = min(dp[v], dp[u]); // 把dp[v]更新为本条路最低价格
if(--in[v] == 0) q.push(v); // 如果v的子结点遍历完了,就可以把v加入队列了
}
}
cout << ans << endl;
}