Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- #include <iostream>
- using namespace std;
- const int maxn = 1e5 + 10;
- int idx[maxn];
- int sz[maxn];
- void init() {
- for(int i = 0; i < maxn; i++) {
- idx[i] = i;
- sz[i] = 1;
- }
- }
- int find_root(int A) {
- while(idx[A] != A) {
- idx[A] = idx[idx[A]];
- A = idx[A];
- }
- return A;
- }
- void unite(int A, int B) {
- int root_A = find_root(A);
- int root_B = find_root(B);
- if(root_A != root_B) {
- if(sz[root_A] < sz[root_B]) {
- sz[root_B] += sz[root_A];
- idx[root_A] = idx[root_B];
- }
- else {
- idx[root_B] = idx[root_A];
- sz[root_A] += sz[root_B];
- }
- }
- }
- bool check_if_they_belong_to_same_set(int A, int B) {
- return find_root(A) == find_root(B);
- }
- int main()
- {
- init();
- while(true) {
- string s;
- cin >> s;
- if(s == "end") {
- break;
- }
- if(s == "union") {
- int a, b;
- cin >> a >> b;
- unite(a, b);
- }
- else {
- int a, b;
- cin >> a >> b;
- cout << check_if_they_belong_to_same_set(a, b) << endl;
- }
- }
- }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement
