Skip to content

Commit

Permalink
Merge pull request #544 from aditmehta9/patch-4
Browse files Browse the repository at this point in the history
Prim's Minimum spanning Tree code in C++
  • Loading branch information
ambujraj authored Oct 7, 2018
2 parents 6277415 + 4f12a89 commit 1aa1602
Showing 1 changed file with 65 additions and 0 deletions.
65 changes: 65 additions & 0 deletions primsminimumspannigtree.cpp
Original file line number Diff line number Diff line change
@@ -0,0 +1,65 @@
#include <stdio.h>
#include <limits.h>
#include<stdbool.h>

#define V 5

int minKey(int key[], bool mstSet[])
{
int min = INT_MAX, min_index;

for (int v = 0; v < V; v++)
if (mstSet[v] == false && key[v] < min)
min = key[v], min_index = v;

return min_index;
}

int printMST(int parent[], int n, int graph[V][V])
{
printf("Edge \tWeight\n");
for (int i = 1; i < V; i++)
printf("%d - %d \t%d \n", parent[i], i, graph[i][parent[i]]);
}

void primMST(int graph[V][V])
{
int parent[V];

int key[V];

bool mstSet[V];

for (int i = 0; i < V; i++)
key[i] = INT_MAX, mstSet[i] = false;

key[0] = 0;
parent[0] = -1;
for (int count = 0; count < V-1; count++)
{
int u = minKey(key, mstSet);

mstSet[u] = true;

for (int v = 0; v < V; v++)

if (graph[u][v] && mstSet[v] == false && graph[u][v] < key[v])
parent[v] = u, key[v] = graph[u][v];
}

printMST(parent, V, graph);
}

int main()
{

int graph[V][V] = {{0, 2, 0, 6, 0},
{2, 0, 3, 8, 5},
{0, 3, 0, 0, 7},
{6, 8, 0, 0, 9},
{0, 5, 7, 9, 0}};

primMST(graph);

return 0;
}

0 comments on commit 1aa1602

Please sign in to comment.