Cod sursa(job #2905035)

Utilizator moltComan Calin molt Data 19 mai 2022 08:55:12
Problema Paduri de multimi disjuncte Scor 70
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.12 kb
#include <bits/stdc++.h>

using namespace std;

#define NMAX 100005

ifstream in("disjoint.in");
ofstream out("disjoint.out");

int parent[NMAX];

int find_parent(int x, vector<int> path) {
    while (x != parent[x]) {
        path.push_back(x);
        x = parent[x];
    }
    return x;
}

void unite(int x, int y) {
    vector<int> path_x, path_y;
    int parent_x = find_parent(parent[x], path_x);
    int parent_y = find_parent(parent[y], path_y);

    parent[parent_x] = parent_y;
    for (int i = 0; i < (int)path_x.size(); ++i) {
        parent[path_x[i]] = parent_y;
    }
}

int main()
{
    int n, m;
    in >> n >> m;

    for (int i = 0; i < n + 1; ++i) {
        parent[i] = i;
    }

    for (int i = 0; i < m; ++i) {
        int op, x, y;
        in >> op >> x >> y;
        if (op == 1) {
            unite(x, y);
        } else {
            vector<int> path_x, path_y;
            int parent_x = find_parent(x, path_x);
            int parent_y = find_parent(y, path_y);
            if (parent_x == parent_y) {
                out << "DA\n";
            } else {
                out << "NU\n";
            }
        }
    }
}