Cod sursa(job #2948457)

Utilizator simonatudoroiuTudoroiu Simona simonatudoroiu Data 27 noiembrie 2022 18:48:24
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.77 kb
#include <bits/stdc++.h>
using namespace std;
ifstream fin("disjoint.in");
ofstream fout("disjoint.out");
//functie de cautat radacinile arborelui din care face parte un nod
int gaseste_radacina(int nod, vector<int>& tata)
{
    //daca este radacina, il returnam
    if(tata[nod] == nod)
        return nod;
    else
    {
        int c = nod;
        vector<int> temp;
        while(tata[c] != c)
        {
            c = tata[c];
//            temp.push_back(c);
        }

//        for(int i:temp)
//        {
//            tata[i] = c;
//        }

        return c;
    }
}
int main()
{
    int n,m,a,b,c;
    vector<vector<int>> list;
    vector<int> tata, rang;
    fin>>n>>m;
    //initializare
    for(int i=0;i<=n;i++)
    {
        tata.push_back(i);
        rang.push_back(1);
        list.push_back({});
    }
    for(int x=1;x<=m;x++)
    {
        fin>>a>>b>>c;
        if(a == 1)
        {
            //gasim radacinile pentru b si c
            int rad_b = gaseste_radacina(b,tata);
            int rad_c = gaseste_radacina(c,tata);
            //comparam inaltimea arborilor ce au ca radacina radacinile nodurilor b si c
            if(rang[rad_b]<rang[rad_c])
                tata[rad_b] = rad_c;
            else
                tata[rad_c] = rad_b;
            //daca au inailimile la fel, actualizam inaltimea radacinei lui c
            if(rang[rad_b] == rang[rad_c])
                rang[rad_c]++;
        }
        else
        {
            //b si c se afla in aceeasi multime daca au aceeasi radacina
            if(gaseste_radacina(b,tata) == gaseste_radacina(c,tata))
                fout<<"DA\n";
            else
                fout<<"NU\n";
        }

    }
    fin.close();
    fout.close();
    return 0;
}