-
Notifications
You must be signed in to change notification settings - Fork 11
Expand file tree
/
Copy pathPrim.cpp
More file actions
101 lines (80 loc) · 1.69 KB
/
Prim.cpp
File metadata and controls
101 lines (80 loc) · 1.69 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
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
// algoritimo de prim
// 1 - definir a distancia de cada vertice como infinito (similar ao dijkstra).
// 2 - definir a distancia de 0 para o source(0).
// 3 - Em cada passo, encontrar o vértice u, que ainda não foi processado, que possua a menor das distâncias.
// 4 - ao termino fazer a soma de todas as distancias e encontrar qual a soma das distancias na MST.
#include <bits/stdc++.h>
using namespace std;
#define lli long long int
#define pb push_back
#define pii pair<int, int>
#define mp make_pair
#define MAXN 100001
#define INF 999999
#define sec second
#define fir first
int n, m, a, b, c;
vector<pii> adj[MAXN];
int dist[MAXN];
bool processed[MAXN];
void prim()
{
for (int i = 0; i < n; i++)
{
dist[i] = INF;
}
dist[0] = 0;
priority_queue<pii, vector<pii>, greater<pii>> q;
q.push(pii(dist[0], 0));
while (1)
{
int davez = -1;
while (!q.empty())
{
int atual = q.top().sec;
q.pop();
if (!processed[atual])
{
davez = atual;
break;
}
}
if (davez == -1)
{
break;
}
processed[davez] = true;
for (int i = 0; i < adj[davez].size(); i++)
{
int distt = adj[davez][i].fir;
int atual = adj[davez][i].sec;
if (dist[atual] > distt && !processed[atual])
{
dist[atual] = distt;
q.push(pii(dist[atual], atual));
}
}
}
int ans = 0;
for (int i = 0; i < n; i++)
{
ans += dist[i];
}
cout << ans << endl;
}
int main()
{
ios_base::sync_with_stdio(false);
cin.tie(NULL);
cin >> n >> m;
for (int i = 0; i < m; i++)
{
cin >> a >> b >> c;
a--;
b--;
adj[a].pb(mp(c, b));
adj[b].pb(mp(c, a));
}
prim();
return 0;
}