forked from k-samarth/Data-structures
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathPrims_cpp
More file actions
77 lines (73 loc) · 1.5 KB
/
Copy pathPrims_cpp
File metadata and controls
77 lines (73 loc) · 1.5 KB
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
#include <iostream>
#include<climits>
using namespace std;
int findMinVertex(bool *visited,int *weights,int n)
{
int min=-1;
for(int i=0;i<n;i++)
{
if(!visited[i] && (min==-1 || weights[i]<weights[min]))
{
min=i;
}
}
return min;
}
void prims(int **edges,int n)
{
int *parent =new int[n];
int *weights =new int[n];
bool*visited =new bool[n];
for(int i=1;i<n;i++)
{
visited[i]=false;
weights[i]=INT_MAX;
}
parent[0]=-1;
weights[0]=0;
for(int i=0;i<n-1;i++)
{
int minVertex=findMinVertex(visited,weights,n);
visited[minVertex]=true;
for(int j=0;j<n;j++)
{
if(!visited[j] && edges[minVertex][j]!=0)
{
if(edges[minVertex][j]<weights[j])
{
weights[j]=edges[minVertex][j];
parent[j]=minVertex;
}
}
}
}
for(int i=1;i<n;i++)
{ if(parent[i]<i)
cout<<parent[i]<<" "<<i<<" "<<weights[i]<<endl;
else
cout<<i<<" "<<parent[i]<<" "<<weights[i]<<endl;
}
}
int main()
{ int f,s,weight;
int n,e;
cin>>n>>e;
int** edges = new int*[n];
for(int i=0;i<n;i++)
{
edges[i]=new int[n];
for(int j=0;j<n;j++)
{
edges[i][j]=0;
}
}
for(int j=0;j<e;j++)
{
cin>>f>>s>>weight;
edges[f][s]=weight;
edges[s][f]=weight;
}
//cout<<edges[1][1];
cout<<endl;
prims(edges,n);
}