Cod sursa(job #2942749)

Utilizator simonatudoroiuTudoroiu Simona simonatudoroiu Data 20 noiembrie 2022 00:04:49
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.3 kb
#include <bits/stdc++.h>
using namespace std;
ifstream fin("disjoint.in");
ofstream fout("disjoint.out");
int gaseste_radacina(int nod, vector<int>& tata)
{
    if(tata[nod] == nod)
        return nod;
    else
    {
        int c = nod, x;
        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;
    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)
        {
            int rad_b = gaseste_radacina(b,tata);
            int rad_c = gaseste_radacina(c,tata);
             if(rang[rad_b]<rang[rad_c])
                tata[rad_b] = rad_c;
            else
                tata[rad_c] = rad_b;
            if(rang[rad_b] == rang[rad_c])
                rang[rad_c]++;
        }
        else
        {

            if(gaseste_radacina(b,tata) == gaseste_radacina(c,tata))
                fout<<"DA\n";
            else
                fout<<"NU\n";
        }

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