Psl/kruskal
รุ่นแก้ไขเมื่อ 07:15, 29 มกราคม 2567 โดย Jittat (คุย | มีส่วนร่วม) (สร้างหน้าด้วย "<syntaxhighlight lang="cpp"> #include <iostream> #include <vector> #include <set> #include <algorithm> using namespace std; const int MAX_N = 100010;...")
#include <iostream>
#include <vector>
#include <set>
#include <algorithm>
using namespace std;
const int MAX_N = 100010;
int n,m;
vector<int> adj[MAX_N];
vector<int> weights[MAX_N];
vector<pair<int,int>> edges;
vector<int> edge_weights;
int deg[MAX_N];
set<pair<int,int>> Q;
vector<pair<int,int>> sedges;
void read_input()
{
cin >> n >> m;
for(int u=0; u<n; u++) deg[u] = 0;
for(int i=0; i<m; i++) {
int a,b,w;
cin >> a >> b >> w; a--; b--;
adj[a].push_back(b);
adj[b].push_back(a);
weights[a].push_back(w);
weights[b].push_back(w);
deg[a]++;
deg[b]++;
edges.push_back(make_pair(a,b));
edge_weights.push_back(w);
sedges.push_back(make_pair(w,i));
}
}
/*
int set_number[MAX_N];
void init_set()
{
for(int i=0; i<n; i++)
set_number[i] = i;
}
int set_find(int u)
{
return set_number[u];
}
void set_union(int pu, int pv)
{
for(int i=0; i<n; i++)
if(set_number[i] == pu)
set_number[i] = pv;
}
*/
int set_parent[MAX_N];
int set_rank[MAX_N];
void init_set()
{
for(int i=0; i<n; i++) {
set_parent[i] = i;
set_rank[i] = 1;
}
}
int set_find(int u)
{
if(set_parent[u] == u) {
return u;
} else {
return set_find(set_parent[u]);
}
}
void set_union(int pu, int pv)
{
pu = set_find(pu);
pv = set_find(pv);
if(pu == pv)
return;
if(set_rank[pv] > set_rank[pu]) {
set_parent[pu] = pv;
} else if(set_rank[pu] > set_rank[pv]) {
set_parent[pv] = pu;
} else {
set_parent[pv] = pu;
set_rank[pu]++;
}
}
main()
{
int total = 0;
read_input();
init_set();
sort(sedges.begin(), sedges.end());
for(int i=0; i<m; i++) {
int e = sedges[i].second;
int u = edges[e].first;
int v = edges[e].second;
int w = edge_weights[e];
int pu = set_find(u);
int pv = set_find(v);
if(pu != pv) {
total += w;
set_union(pu,pv);
}
}
cout << total << endl;
}