Cod sursa(job #2947423)

Utilizator VladTalpigaVlad Talpiga VladTalpiga Data 26 noiembrie 2022 02:08:26
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.8 kb
#include <iostream>
#include <fstream>
#include <vector>

using namespace std;

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

vector <int> tata(100001);

int root(int node)  //urcam recursiv in arbore pana gasim radacina
{
    if(tata[node] == node) //am gasit radacina arborelui
        return node;

    return root(tata[node]);
}

int main()
{
    int n, m, i, cod, x, y;

    f >> n >> m;
    tata.resize(n+1);

    for(i = 1; i <= n; i++)
        tata[i] = i;

    for(i = 1; i <= m; i++)
    {
        f >> cod >> x >> y;

        if(cod == 1)
            tata[root(y)] = root(x);
        else if(root(x) == root(y)) //daca au aceeasi radacina => sunt in acelasi arbore
            g << "DA\n";
        else
            g << "NU\n";
    }

    return 0;
}