代码随想录算法训练营第六十天 | 图论part10

94. 城市间货物运输 I

对于Bellman_ford算法的优化,松弛n-1次,并且每一次都松弛每一条边,其实做了许多没有意义的事情。实际上只去松弛上一次计算过的节点作为出发节点的边即可。

cpp 复制代码
#include <iostream>
#include <vector>
#include <list>
#include <queue>
#include <fstream>
#include <climits>

using namespace std;

struct Edge {
	int end, val;
	Edge(int end, int val) :end(end), val(val) {}
};

int main() {
	int n, m;
	int s, e, v;
	ifstream infile("input.txt");
	cin >> n >> m;
	vector<list<Edge>> graph(n + 1);
	
	while (m--) {
		cin >> s >> e >> v;
		graph[s].emplace_back(Edge(e, v));
	}
	
	// 记录到源点的距离的一维数组  
	vector<int> minDist(n + 1, INT_MAX);
	// 记录上一次计算的节点
	queue<int> que;

	minDist[1] = 0;
	que.push(1);
	
	while (!que.empty()) {
		int cur = que.front();
		que.pop();

		for (Edge e : graph[cur]) {
			if (minDist[e.end] > minDist[cur] + e.val) {
				minDist[e.end] = minDist[cur] + e.val;
				que.push(e.end);
			}
		}
		
	}
	
	if (minDist[n] == INT_MAX) cout << "unconnected" << endl;
	else cout << minDist[n] << endl;
	return 0;
}

95. 城市间货物运输 II

只需要记录每个节点入队列的次数,超过n-1直接输出circle然后返回。

cpp 复制代码
#include <iostream>
#include <vector>
#include <list>
#include <queue>
#include <fstream>
#include <climits>

using namespace std;

struct Edge {
	int end, val;
	Edge(int end, int val) :end(end), val(val) {}
};

int main() {
	int n, m;
	int s, e, v;
	ifstream infile("input.txt");
	cin >> n >> m;
	vector<list<Edge>> graph(n + 1);
	
	while (m--) {
		cin >> s >> e >> v;
		graph[s].emplace_back(Edge(e, v));
	}
	
	// 记录到源点的距离的一维数组  
	vector<int> minDist(n + 1, INT_MAX);
	// 记录上一次计算的节点
	queue<int> que;
	// 记录每一个节点入队列的次数,不超过n-1
	vector<int> count(n + 1, 0);

	minDist[1] = 0;
	que.push(1);
	count[1]++;
	
	while (!que.empty()) {
		int cur = que.front();
		que.pop();

		for (Edge e : graph[cur]) {
			if (minDist[e.end] > minDist[cur] + e.val) {
				minDist[e.end] = minDist[cur] + e.val;
				que.push(e.end);
				count[e.end]++;
				if (count[e.end] > n - 1) {
					cout << "circle" << endl;
					return 0;
				}
			}
		}
		
	}
	
	if (minDist[n] == INT_MAX) cout << "unconnected" << endl;
	else cout << minDist[n] << endl;
	return 0;
}

96. 城市间货物运输 III

对所有边松弛一次,相当于计算 起点到达与起点同一条边的节点的最短距离。

所以应该最多松弛k+1条边。并且每次松弛只能使用上次结果的副本。

cpp 复制代码
#include <iostream>
#include <vector>
#include <list>
#include <queue>
#include <fstream>
#include <climits>

using namespace std;

struct Edge {
	int source, end, val;
	Edge(int source, int end, int val) :source(source), end(end), val(val) {}
};

int main() {
	int n, m;
	int s, e, v;
	int src, dst, k;
	ifstream infile("input.txt");
	cin >> n >> m;
	vector<Edge> graph;
	
	while (m--) {
		cin >> s >> e >> v;
		graph.emplace_back(Edge{s, e, v});
	}
	cin >> src >> dst >> k;

	// 记录到源点的距离的一维数组  
	vector<int> minDist(n + 1, INT_MAX);
	// 上一轮的副本
	vector<int> minDistCopy(n + 1);

	minDist[src] = 0;
	
	for (int i = 0; i <= k; ++i) { // 重复k+1次
		minDistCopy = minDist;

		for (Edge edge : graph) {
			if (minDistCopy[edge.source] != INT_MAX && minDist[edge.end] > minDistCopy[edge.source] + edge.val) {
				minDist[edge.end] = minDistCopy[edge.source] + edge.val;
			}
		}
	}
	
	
	if (minDist[dst] == INT_MAX) cout << "unreachable" << endl;
	else cout << minDist[dst] << endl;
	return 0;
}
相关推荐
艾莉丝努力练剑1 小时前
【LeetCode&数据结构】单链表的应用——反转链表问题、链表的中间节点问题详解
c语言·开发语言·数据结构·学习·算法·leetcode·链表
_殊途3 小时前
《Java HashMap底层原理全解析(源码+性能+面试)》
java·数据结构·算法
珊瑚里的鱼6 小时前
LeetCode 692题解 | 前K个高频单词
开发语言·c++·算法·leetcode·职场和发展·学习方法
秋说7 小时前
【PTA数据结构 | C语言版】顺序队列的3个操作
c语言·数据结构·算法
lifallen8 小时前
Kafka 时间轮深度解析:如何O(1)处理定时任务
java·数据结构·分布式·后端·算法·kafka
liupenglove8 小时前
自动驾驶数据仓库:时间片合并算法。
大数据·数据仓库·算法·elasticsearch·自动驾驶
python_tty9 小时前
排序算法(二):插入排序
算法·排序算法
然我9 小时前
面试官:如何判断元素是否出现过?我:三种哈希方法任你选
前端·javascript·算法
F_D_Z9 小时前
【EM算法】三硬币模型
算法·机器学习·概率论·em算法·极大似然估计
秋说10 小时前
【PTA数据结构 | C语言版】字符串插入操作(不限长)
c语言·数据结构·算法