题目描述
题目分析
用于练习计算负权值边最短路径的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;
}