-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathMST.cpp
More file actions
68 lines (61 loc) · 1.25 KB
/
MST.cpp
File metadata and controls
68 lines (61 loc) · 1.25 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
#include<bits/stdc++.h>
using namespace std;
#define ll long long
bool compare(vector<int> &a, vector<int>&b){
return a[2] < b[2];
}
int findParent(int u, vector<int>&parent){
if(parent[u]==u){
return u;
}
return parent[u] = findParent(parent[u],parent);
}
void Union(int u, int v, vector<int>&rank, vector<int>&parent){
int parU = findParent(u,parent);
int parV = findParent(v,parent);
if(parU == parV){
return ;
}
if(rank[parU] > rank[parV]){
parent[parV] = parent[parU];
}
else if(rank[parU] < rank[parV]){
parent[parU] = parent[parV];
}
else{
parent[parU] = parent[parV];
rank[parV]++;
}
}
int main(){
int nodes;
cin>>nodes;
int edgesCount;
cin>>edgesCount;
vector<vector<int>>graph(nodes);
vector<vector<int>>edges;
for(int i=0;i<edgesCount;i++){
int sou,des,wt;
cin>>sou>>des>>wt;
edges.push_back({sou,des,wt});
graph[sou].push_back(des);
graph[des].push_back(sou);
}
vector<int>parent(nodes);
vector<int>rank(nodes,0);
for(int i=0;i<nodes;i++){
parent[i]=i;
}
sort(edges.begin(),edges.end(), compare);
int cost = 0;
for(int i=0;i<edges.size();i++){
int sou = edges[i][0];
int des = edges[i][1];
if(parent[sou]!=parent[des]){
cost+=edges[i][2];
Union(sou,des,rank,parent);
}
}
cout<<cost;
return 0;
}