minimario

<Kruskal> TEMPLATE

Oct 9th, 2015
138
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 1.82 KB | None | 0 0
  1. #include <bits/stdc++.h>
  2. using namespace std;
  3.  
  4. #define ll long long
  5. #define mp make_pair
  6. #define pb push_back
  7. #define pii pair <int, int>
  8. #define piii pair<pii, int>
  9. #define vi vector<ll>
  10.  
  11. #define read1(a) int a; scanf("%d", &a)
  12. #define read2(a, b) int a, b; scanf("%d %d", &a, &b)
  13. #define read3(a, b, c) int a, b, c; scanf("%d %d %d", &a, &b, &c)
  14.  
  15. #define FOR(i, a, b) for (int i=a; i<b; i++)
  16. #define F0R(i, a) for (int i=0; i<a; i++)
  17.  
  18. #define readgi(n) F0R(i, n) { scanf("%d", &arr[i]); }
  19. #define readgs(n) F0R(i, n) { scanf(" %c", &arr[i]); }
  20.  
  21. #define f first
  22. #define s second
  23.  
  24. #define usaco(in, out) freopen(in, "r", stdin); freopen(out, "w", stdout);
  25.  
  26. #define println1(a) printf("%d\n", a);
  27. #define println2(a, b) printf("%d %d\n", a, b);
  28. #define println3(a, b, c) printf("%d %d %d\n", a, b, c);
  29. #define pv(v) for (int i : v) { printf("%d ", i); } printf("\n");
  30.  
  31. const int MOD = 1000000007;
  32. const int MAX = 100005;
  33.  
  34. int arr[MAX];
  35. int idx[MAX];
  36. int sze[MAX];
  37.  
  38. int root(int x) {
  39.     while (x != idx[x]) {
  40.         idx[x] = idx[idx[x]];
  41.         x = idx[x];
  42.     }
  43.     return x;
  44. }
  45.  
  46. bool conn(int p, int q) {
  47.     return root(p) == root(q);
  48. }
  49.  
  50. void join(int p, int q) {
  51.     int i = root(p);
  52.     int j = root(q);
  53.     if (sze[i] < sze[j]) {
  54.         idx[i] = j;
  55.         sze[j] += sze[i];
  56.     }
  57.  
  58.     else {
  59.         idx[j] = i;
  60.         sze[i] += sze[j];
  61.     }
  62. }
  63.  
  64. vector<pair<int, pii> > edges;
  65.  
  66. int main() {
  67.     read2(n, m);
  68.     F0R(i, n) { idx[i] = i; sze[i] = 1; }
  69.     F0R(i, m) {
  70.         read3(x, y, w);
  71.         edges.pb(mp(w, mp(x, y)));
  72.     }
  73.  
  74.     sort(edges.begin(), edges.end());
  75.  
  76.     int mst = 0;
  77.     for (pair<int, pii> i : edges) {
  78.         if (!conn(i.s.f, i.s.s)) {
  79.             mst += i.f;
  80.             join(i.s.f, i.s.s);
  81.         }
  82.     }
  83.  
  84.     cout << mst << endl;
  85.     return 0;
  86. }
Advertisement
Add Comment
Please, Sign In to add comment