Bellman-Ford 算法

/**
该代码修改自《挑战程序设计》第二版
*/
#include <iostream>
#include <string.h>
#include <stack>
#define INF 1000000000
#define MAX_E 99
#define MAX_V 100
using namespace std;
struct edge{
int  from,to,cost;//边的起点，终点，权值
};
edge es[MAX_E];
int d[MAX_V];//源点到顶点的最短距离
int path[MAX_V];//存最短路径
int V,E;//顶点数，边数
int s;//源点
void BellmanFord(int s);
bool find_negative_loop();//判断是否有负环
void Print();
int main()
{
    cout << "请输入图的顶点数，边数，源点：";
    cin >>V>>E>>s;

    cout << "请输入" <<E<< "条边的起点、终点以及权值：\n";
    for (int i = 0; i <E; i++)
        cin >>es[i].from>> es[i].to >> es[i].cost;
    if(!find_negative_loop()){
        BellmanFord(s);
        Print();
    }else{
    cout<<"有负环"<<endl;
    }

    return 0;
}
void BellmanFord(int s){
    for(int i=0;i<V;i++){
        d[i]=INF;
    }
    d[s]=0;
    while(true){//最多循环V-1次
        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;
                path[e.to]=e.from;
                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;//如果其在第V次仍然更新则说明其有负环
            ///因为最短路径不可能经过同一个点两次，所以最多循环V-1次
            ///但如果有负环会无限循环下去，自己可以举个栗子试一下
        }
        }
    }
    return false;

}

void Print()
{
    for (int i = 0; i < V; i++)
    {
        if (i != s)
        {
            int p = i;
            stack<int> st;
            cout << "顶点 " << s << " 到顶点 " << p << " 的最短路径是： ";

            while (s != p)  //路径顺序是逆向的，所以先保存到栈
            {
                st.push(p);
                p = path[p];
            }

            cout << s;
            while (!st.empty())  //依次从栈中取出的才是正序路径
            {
                cout << "--" << st.top();
                st.pop();
            }
            cout << "    最短路径长度是：" << d[i] << endl;
        }

    }
}