Cod sursa(job #2938448)

Utilizator SteanfaDiaconu Stefan Steanfa Data 12 noiembrie 2022 09:17:02
Problema Paduri de multimi disjuncte Scor 70
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.82 kb
#include <iostream>
#include <fstream>
#include <vector>

std::vector<int> parint(100001);

int fetch(int val)
{
    if (parint[val] == val)
    {
        return val;
    }
    return fetch(parint[val]);
}
void reune(int casa, int masa)
{
    int a = fetch(casa);
    int b = fetch(masa);
    parint[a] = b;
}

int main()
{
    std::ifstream cin("disjoint.in");
    std::ofstream cout("disjoint.out");

    int n, m;
    cin >> n >> m;
    int q, first, sec;
    for (size_t i = 1; i <= n; i++)
    {
        parint[i] = i;
    }

    for (size_t i = 0; i < m; i++)
    {
        cin >> q >> first >> sec;
        if (q == 1)
        {
            reune(first, sec);
        }
        else
        {
            if (fetch(first) == fetch(sec))
                cout << "DA\n";
            else
                cout << "NU\n";
        }
    }
}