Cod sursa(job #2819985)

Utilizator TudosieRazvanTudosie Marius-Razvan TudosieRazvan Data 19 decembrie 2021 16:08:05
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.17 kb
#include <fstream>

using namespace std;


ifstream fin ("disjoint.in");
ofstream fout ("disjoint.out");

int n,m;
int tata[100001];

int cRad(int frunza)
{
    int rad=frunza;
    while(tata[rad]>0)
    {
        rad=tata[rad];
    }
    return rad;
}

int main()
{
    fin>>n>>m;
    for(int i=1; i<=n; i++)
    {
        tata[i]=-1;
    }
    for(int i=1; i<=m; i++)
    {
        int cer;
        fin>>cer;
        int f1,f2;
        fin>>f1>>f2;
        int rad1=cRad(f1);
        int rad2=cRad(f2);
        if(cer==1)
        {
            //unific arborii


            if(rad1!=rad2)
            {
                //am radacini dif
                if(tata[rad1]<tata[rad2])
                {
                    tata[rad1]+=tata[rad2];
                    tata[rad2]=rad1;
                }
                else{
                    tata[rad2]+=tata[rad1];
                    tata[rad1]=rad2;
                }
            }
        }
        else{
            if(rad1==rad2)
            {
                fout<<"DA\n";
            }
            else{
              fout<<"NU\n";
            }
        }
    }
    return 0;
}