Advertisement
minimario

<DSU> TEMPLATE

Nov 25th, 2015
102
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 1.46 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. #define vpii vector<pii>
  11.  
  12. #define read1(a) int a; scanf("%d", &a)
  13. #define read2(a, b) int a, b; scanf("%d %d", &a, &b)
  14. #define read3(a, b, c) int a, b, c; scanf("%d %d %d", &a, &b, &c)
  15.  
  16. #define FOR(i, a, b) for (int i=a; i<b; i++)
  17. #define F0R(i, a) for (int i=0; i<a; i++)
  18.  
  19. #define readgi(n) F0R(i, n) { scanf("%d", &arr[i]); }
  20. #define readgs(n) F0R(i, n) { scanf(" %c", &arr[i]); }
  21.  
  22. #define f first
  23. #define s second
  24.  
  25. #define usaco(in, out) freopen(in, "r", stdin); freopen(out, "w", stdout);
  26.  
  27. #define println1(a) printf("%d\n", a);
  28. #define println2(a, b) printf("%d %d\n", a, b);
  29. #define println3(a, b, c) printf("%d %d %d\n", a, b, c);
  30. #define pv(v) for (int i : v) { printf("%d ", i); } printf("\n");
  31.  
  32. const int MOD = 1000000007;
  33. const int MAX = 100005;
  34.  
  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 connected(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. int main() {
  65.     F0R(i, MAX) { idx[i] = i; sze[i] = 1; }
  66. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement