【2024.3.10练习】有边数限制的最短路

题目描述

---

题目分析

用于练习计算负权值边最短路径的Bellman-Ford算法。看似板子题，但要做到限制步数仍有难度。因为Bellman-Ford算法没有记录步数的功能，且由于存在回路，计算中会反复覆盖同一个结点信息。所以利用动态数组存储不能覆盖的结点信息。

---

我的代码

然而这段代码时间复杂度太大，一旦额外存储的结点信息过多就会运行超时。

#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
int n; //结点数
int m; //边数
int k; //最大步数
pair<pair<int, int>, int> cost[10002]; //顶点间的边权值 
vector<pair<int, int>> V[502]; //记录最短路径长度和所需步数的向量的数组
vector<pair<int, int>>::iterator it; //向量迭代器
vector<pair<int, int>>::iterator it2; //向量迭代器
int main() {
	/*初始化*/
	cin >> n >> m >> k;
	int x;
	int y;
	int z;
	pair<int, int> P(0, 0);
	V[1].insert(V[1].begin(), P);
	for (int i = 2; i <= n; i++)
	{
		pair<int, int> P(10000000, 1000);
		V[i].insert(V[i].begin(),P); //录入最初结点信息
	}
	for (int i = 1; i <= m; i++) {
		cin >> x >> y >> z;
		pair<int, int> P1(x, y);
		pair<pair<int, int>, int> P2(P1, z);
		cost[i] = P2; //输入边权值
	}
	/*Bellman-Ford算法*/
	for (int i = 1; i <= k; i++) { //总遍历k次
		for (int j = 1; j <= m; j++) //每次遍历取出所有边
		{
			int s = cost[j].first.first;
			int t = cost[j].first.second;
			int v = cost[j].second; //取出两结点、边
			for (it = V[s].begin(); it < V[s].end(); it++) //循环取出s结点中储存的信息
			{
				int s_len = (*it).first;
				int s_step = (*it).second;
				for (it2 = V[t].begin(); it2 < V[t].end(); it2++) //循环取出结点中储存的信息
				{
					int t_len = (*it2).first;
					int t_step = (*it2).second;
					/*1.距离更短，且步数不会更多:覆盖结点信息*/
					if(s_step + 1 == i){
					if (s_len + v < t_len && s_len!= 10000000 && s_step + 1 <= t_step) {
						it2 = V[t].erase(it2); //返回删除元素后一个元素位置
						pair<int, int> P(s_len + v, s_step + 1);
						it2 = V[t].insert(it2, P); //在指定元素前添加元素，返回所添加元素的位置
					}
					/*2.距离更短，且步数更多:额外储存结点信息*/
					else if (s_len + v < t_len && s_len != 10000000 && s_step + 1 > t_step) {
						pair<int, int> P(s_len + v, s_step + 1);
						it2 = V[t].insert(it2+1, P); //添加元素
					}
					/*3.距离更长，且步数更少:额外储存结点信息*/
					else if (s_len + v > t_len && s_len != 10000000 && s_step + 1 < t_step) {
						pair<int, int> P(s_len + v, s_step + 1);
						it2 = V[t].insert(it2+1, P); //添加元素
					}
					}
				}
			}
		}
	}
	/*获取答案*/
	int ans = 10000000;
	for (it = V[n].begin(); it < V[n].end(); it++) {
		int num = (*it).first;
		ans = min(num, ans);
	}
	if (ans == 10000000) {
		cout << "impossible";
	}
	else {
		cout << ans;
	}
	return 0;
}