Advertisement
esraa_syam

B. Disjoint Sets Union 2

Mar 24th, 2023
120
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 3.45 KB | None | 0 0
  1. #include<bits/stdc++.h>
  2. #define nl "\n"
  3. #define fi first
  4. #define se second
  5. #define pi 3.14159
  6. #define ll long long
  7. #define odd(a)  (a&1)
  8. #define even(a)  !(a&1)
  9. #define Mod 1'000'000'007
  10. #define INF 2'000'000'000
  11. #define sz(x) int(x.size())
  12. #define charToInt(s) (s - '0')
  13. #define ull unsigned long long
  14. #define number_line iota(all(vec) , 1)
  15. #define all(s) s.begin(), s.end()
  16. #define rall(v) v.rbegin() , v.rend()
  17. #define fixed(n) fixed << setprecision(n)
  18. #define Num_of_Digits(n) ((int)log10(n) + 1)
  19. #define to_decimal(bin) stoll(bin, nullptr, 2)
  20. #define Ceil(n, m) (((n) / (m)) + ((n) % (m) ? 1 : 0))
  21. #define Floor(n, m) (((n) / (m)) - ((n) % (m) ? 0 : 1))
  22. #define Upper(s) transform(all(s), s.begin(), ::toupper);
  23. #define Lower(s) transform(all(s), s.begin(), ::tolower);
  24. #define cout_map(mp) for(auto& [f, s] : mp) cout << f << "  " << s << "\n";
  25. //  ----- bits-------
  26. #define pcnt(n) __builtin_popcount(n)  
  27. #define pcntll(n) __builtin_popcountll(n)
  28. #define clz(n) __builtin_clz(n)    // <---100
  29. #define clzll(n) __builtin_clzll(n)
  30. #define ctz(n) __builtin_ctz(n)  // 0001---->
  31. #define ctzll(n) __builtin_ctzll(n)
  32.  
  33. using namespace std;
  34.  
  35. template < typename T = int > istream& operator >> (istream &in, vector < T > & v){
  36.     for(auto & x : v) in >> x;
  37.     return in;
  38. }
  39.  
  40. template < typename T = int > ostream& operator << (ostream &out, const vector < T > & v){
  41.     for(const T & x : v) out << x << nl;
  42.     return out;
  43. }
  44.  
  45. void esraa()
  46. {
  47.     //freopen("shuffle.in", "r", stdin), freopen("shuffle.out", "w", stdout);
  48.     ios_base::sync_with_stdio(false), cin.tie(nullptr), cout.tie(nullptr);
  49. #ifndef ONLINE_JUDGE
  50.     freopen("input.txt", "r", stdin), freopen("output.txt", "w", stdout);
  51. #endif
  52. }
  53. int maxi = 0;
  54. vector < set < int > > same_set;
  55. struct DSU{
  56.     vector < int > per , rank , group_size;
  57.     int group_count;
  58.     int min_el = 1e9 , max_el = 0;
  59.     explicit DSU(int n){
  60.         per.resize(n + 1);
  61.         rank.resize(n + 1);
  62.         group_size.resize(n + 1);
  63.         group_count = n;
  64.         iota(all(per) , 0);
  65.         fill(all(rank) , 1);
  66.         fill(all(group_size) , 1);
  67.         same_set.resize(n + 1);
  68.     }
  69.  
  70.     int get(int x){
  71.         return (x == per[x] ? x : per[x] = get(per[x]));
  72.     }
  73.  
  74.     void join(int u , int v){
  75.         u = get(u);
  76.         v = get(v);
  77.         if(u == v) return;
  78.         if(rank[u] < rank[v]) swap(u , v);
  79.         per[v] = u;
  80.         group_size[u] += group_size[v];
  81.         maxi = max(maxi , group_size[u]);
  82.         group_count--;
  83.     }
  84.  
  85.     bool is_same_set(int u , int v){
  86.         return get(u) == get(v);
  87.     }
  88. };
  89. void solve(){
  90.     int n , m;
  91.     cin >> n >> m;
  92.     DSU dsu(n);
  93.     for(int i = 0; i < m; i++){
  94.         string s;
  95.         cin >> s;
  96.         if(s == "union"){
  97.             int u , v;
  98.             cin >> u >> v;
  99.             dsu.join(u , v);
  100.             same_set[dsu.get(u)].insert(u);
  101.             same_set[dsu.get(u)].insert(v);
  102.         }
  103.         else{
  104.             int u;
  105.             cin >> u;
  106.             if(!same_set[dsu.get(u)].empty()){
  107.                 cout << *same_set[dsu.get(u)].begin() << " " << *same_set[dsu.get(u)].rbegin() << " " << dsu.group_size[dsu.get(u)] << nl;
  108.             }
  109.             else cout << u << " " << u << " " << dsu.group_size[dsu.get(u)] << nl;
  110.            
  111.            
  112.         }
  113.     }
  114. }
  115. int main()
  116. {
  117.     esraa();
  118.     int t = 1;
  119.     //cin >> t;
  120.     //cin.ignore();
  121.     while(t--)
  122.       solve();
  123.     return 0;
  124. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement