Bellman-ford algorithm to find the shortest path

//
//  main.cpp
//  Demo
//
//  Created by Longxiang Lyu on 8/9/16.
//  Copyright (c) 2016 Longxiang Lyu. All rights reserved.
//

#include <iostream>
#include <string>
#include <memory>
#include <vector>
#include <stdexcept>

class Edge;
class Graph;

using namespace std;

class Edge
{
    friend class Graph;
private:
    int src;
    int dest;
    int weight;
public:
    Edge() = default;
    Edge(int s, int d, int w) : src(s), dest(d), weight(w) {}
};

class Graph
{
    
private:
    int V;
    vector<Edge> *edges;
public:
    Graph() = default;
    Graph(int v) : V(v) { edges = new vector<Edge>(); }
    
    void addEdge(int s, int d, int w)
    {
        edges->push_back(Edge(s, d, w));
    }
    
    void bellmanFord(vector<int> &dist, int src)
    {
        // set up the distance array
        dist.clear();
        dist.resize(V, INT_MAX);
        dist[src] = 0;
        
        // relaxation for |V| - 1 times
        for (int i = 1; i != V; ++i)
        {
            for (auto it = edges->begin(); it != edges->end(); ++it)
            {
                int u = (*it).src;
                int v = (*it).dest;
                int w = (*it).weight;
                if (dist[u] != INT_MAX && dist[u] + w < dist[v])
                    dist[v] = dist[u] + w;
            }
        }
        
        // check for negative cycle
        
        for (auto it = edges->begin(); it != edges->end(); ++it)
        {
            int u = (*it).src;
            int v = (*it).dest;
            int w = (*it).weight;
            if (dist[u] != INT_MAX && dist[u] + w < dist[v])
                throw runtime_error("Negative Weight Cycle Exist!");
        }

    }
};


int main()
{
    Graph graph(5);
    graph.addEdge(0, 1, -1);
    graph.addEdge(0, 2, 4);
    graph.addEdge(1, 2, 3);
    graph.addEdge(1, 3, 2);
    graph.addEdge(1, 4, 2);
    graph.addEdge(3, 2, 5);
    graph.addEdge(3, 1, 1);
    graph.addEdge(4, 3, -3);
    
    vector<int> dist;
    graph.bellmanFord(dist, 0);
    
    for (auto a : dist)
        cout << a << " ";
    cout << endl;

    return 0;
}

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值