Cod sursa(job #2938454)

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

std::vector<int> parint(100001);
std::vector<int> grad(100001, 0);

int fetch(const int &val)
{
    if (parint[val] == val)
    {
        return val;
    }
    return fetch(parint[val]);
}
void reune(const int &casa, const int &masa)
{
    int a = fetch(casa);
    int b = fetch(masa);
    if (grad[a] > grad[b])
    {
        grad[a]++;
        parint[b] = a;
    }
    else
    {
        grad[b]++;
        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";
        }
    }
}