-
Notifications
You must be signed in to change notification settings - Fork 0
/
Bellman_Ford_algorithm.cpp
139 lines (116 loc) · 3.39 KB
/
Bellman_Ford_algorithm.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
//A C++ program for Bellman-Ford's single source
//shortest path algorithm
#include <bits/stdc++.h>
using namespace std;
//a structure to represent a weighted edge in graph
struct Edge {
int src,dest,weight;
};
//a structure to represent a connected,directed and
//weighted graph
struct Graph{
//V->Number of vertices,E->Number of edges
int V,E;
//graph is represented as an array if edges.
struct Edge* edge;
};
//crestes a graph with V vertices and E edges
struct Graph* createGraph(int V,int E){
struct Graph* graph=new Graph;
graph->V=V;
graph->E=E;
graph->edge=new Edge[E];
return graph;
};
//A utility function used to print the solution.
void printArr(int dist[],int n){
printf("Vertex distance from source\n");
for(int i=0;i<n;i++)
printf("%d\t\t%d\n",i,dist[i]);
}
//The main function that finds shortest distances from src
//to all other vertices using Bellman-Ford algorithm.
//function also detects negative weight cycle
void BellmanFord(struct Graph* graph,int src)
{
int V=graph->V;
int E=graph->E;
int dist[V];
//Step 1:Initialize diatances from src to all other
//vertices as INFINITE
for(int i=0;i<V;i++)
dist[i]=INT_MAX;
dist[src]=0;
//Step 2:Relax all edges |V|-1 times.A simple
//shortest path from src to any other vertex can have
//at most |V|-1 edges
for(int i=1;i<=V-1;i++){
for(int j=0;j<E;j++){
int u=graph->edge[j].src;
int v=graph->edge[j].dest;
int weight=graph->edge[j].weight;
if(dist[u]!=INT_MAX&&dist[u]+weight<dist[v])
dist[v]=dist[u]+weight;
}
}
//Step 3: check for negative-weight cycles.The above
//step guarantees shortest distances if graph doesn't
//contain negative weight cycle. If we get a shorter path,
//then there is a negative-weight cycle.
for(int i=0;i<E;i++){
int u=graph->edge[i].src;
int v=graph->edge[i].dest;
int weight=graph->edge[i].weight;
if(dist[u]!=INT_MAX&&dist[u]+weight<dist[v]){
printf("Graph contains negative weight cycle");
return;//If negative cycle is detected,simply
//return
}
}
printArr(dist,V);
return;
}
// Driver's code
int main()
{
/* Let us create the graph given in above example */
int V = 5; // Number of vertices in graph
int E = 8; // Number of edges in graph
struct Graph* graph = createGraph(V, E);
// add edge 0-1 (or A-B in above figure)
graph->edge[0].src = 0;
graph->edge[0].dest = 1;
graph->edge[0].weight = -1;
// add edge 0-2 (or A-C in above figure)
graph->edge[1].src = 0;
graph->edge[1].dest = 2;
graph->edge[1].weight = 4;
// add edge 1-2 (or B-C in above figure)
graph->edge[2].src = 1;
graph->edge[2].dest = 2;
graph->edge[2].weight = 3;
// add edge 1-3 (or B-D in above figure)
graph->edge[3].src = 1;
graph->edge[3].dest = 3;
graph->edge[3].weight = 2;
// add edge 1-4 (or B-E in above figure)
graph->edge[4].src = 1;
graph->edge[4].dest = 4;
graph->edge[4].weight = 2;
// add edge 3-2 (or D-C in above figure)
graph->edge[5].src = 3;
graph->edge[5].dest = 2;
graph->edge[5].weight = 5;
// add edge 3-1 (or D-B in above figure)
graph->edge[6].src = 3;
graph->edge[6].dest = 1;
graph->edge[6].weight = 1;
// add edge 4-3 (or E-D in above figure)
graph->edge[7].src = 4;
graph->edge[7].dest = 3;
graph->edge[7].weight = -3;
// Function call
BellmanFord(graph, 0);
delete [] graph->edge;
return 0;
}