《挑战程序设计竞赛》,初级篇–图
// 单源最短路径1(含负权边) O(V*E)
// Bellman-Ford
#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
#define maxv 101
#define maxe 101
#define INF 9999
struct edge
{
int from,to;
int cost;
};
edge es[maxe];
int d[maxv];
int V,E;
// 从点s出发,到所有点的最短距离
void shortest_path(int s)
{
for(int i = 0; i < V; i++)
d[i] = INF;
d[s] = 0;
while(true)
{
bool update = false;
for(int i = 0; i < E; i++)
{
edge e = es[i];
if(d[e.from] != INF && d[e.to] > d[e.from] + e.cost)
{
d[e.to] = d[e.from] + e.cost;
update = true;
}
}
if(!update) break;
}
}
bool find_negative_loop()
{
memset(d,0,sizeof(d));
for(int i = 0; i < V; i++)
{
for(int j = 0; j < E; j++)
{
edge e = es[j];
if(d[e.to] > d[e.from] + e.cost)
{
d[e.to] = d[e.from] + e.cost;
if(i == V-1) return true;
}
}
}
return false;
}
int main()
{
cin>>V>>E;
for(int i = 0; i < E; i++)
cin>>es[i].from>>es[i].to>>es[i].cost;
shortest_path(0);
printf("------------- \n");
for(int i = 0; i < V; i++)
{
cout<<"该点到"<<i<<"点最短距离为: ";
cout<<d[i]<<endl;
}
return 0;
}
/*
测试数据:
4 4
0 1 1
0 2 3
1 2 4
2 3 2
*/