Cod sursa(job #2941635)

Utilizator Cosmina_GheorgheGheorghe Cosmina Cosmina_Gheorghe Data 18 noiembrie 2022 00:45:32
Problema Paduri de multimi disjuncte Scor 50
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.85 kb
#include <iostream>
#include <fstream>
#include <vector>
using namespace std;
ifstream fin("disjoint.in");
ofstream fout("disjoint.out");
vector <int> tata;
int n,m,tip,x,y;

int radacina(int nod)
{
    //daca nodul nu este radacina o caut la tatal acestuia
    if(nod!=tata[nod])
    {
        return radacina(tata[nod]);
    }
    return nod;
}
int main()
{
    fin>>n>>m;
    tata.resize(n+1);
    for(int i=0;i<n;i++)
    {
        tata[i]=i;
    }
    for(int j=0;j<m;j++)
    {
        fin>>tip>>x>>y;
        if(tip==1)
        {
            //unire arbori
            tata[radacina(x)]=radacina(y);
        }
        else
        {
            if(radacina(x)==radacina(y))
            {
                fout<<"DA";
            }
            else fout<<"NU";
            fout<<endl;
        }
    }
    return 0;
}