bhupixb

DSU

Jun 23rd, 2018
79
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.18 KB | None | 0 0
  1. #include <bits/stdc++.h>
  2. using namespace std;
  3. const int N = 100 * 100 * 10 + 5;
  4. int parent[N], size1[N];
  5. vector< vector < int > > v(20);
  6.  
  7. void make_set(int n, int x = 1){
  8. if(x > n)return;
  9. parent[x] = x;
  10. make_set(n, x + 1);
  11. }
  12. int find(int x){
  13.  
  14. while(x != parent[x])x = parent[x];
  15. return x;
  16.  
  17. }
  18. void merge_set(int x, int y){
  19. int rep_x = find(x);
  20. int rep_y = find(y);
  21.  
  22. if(rep_y == rep_x)return;
  23.  
  24. if(size1[rep_x] > size1[rep_y]){
  25. parent[rep_y] = rep_x;
  26. size1[rep_x] += size1[rep_y];
  27. }
  28.  
  29. else {
  30. parent[rep_x] = rep_y;
  31. size1[rep_y] += size1[rep_x];
  32. }
  33. }
  34. int main()
  35. {
  36. char ch;int x, y, n, q, child;
  37.  
  38. scanf("%d %d", &n, &q);
  39. make_set(n);
  40. for(int i = 1; i <= n; i++)size1[i] = 1;
  41. //for(int i = 1; i < 4; i++)cout << size1[i] << ' ';
  42. //for(int i = 1; i < 4; i++)cout << parent[i] << ' ';
  43.  
  44. for(int i=0; i<q; i++) {
  45. cin >> ch;
  46. if(ch == 'Q'){
  47. cin >> child;
  48. cout << size1[find(child)] << '\n';
  49. }
  50. else{
  51. cin >> x >> y;
  52. merge_set(x, y);
  53. }
  54. }
  55. // for(int i=1; i<=n; i++) if(!seen[i]) dfs(i);
  56. return 0;
  57. }
Add Comment
Please, Sign In to add comment