Cod sursa(job #2946275)

Utilizator anaella2010Chiriac Ella anaella2010 Data 24 noiembrie 2022 18:14:29
Problema Paduri de multimi disjuncte Scor 70
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.11 kb
///disjoint - pentru fiecare operatie 1, voi construi un arbore care sa contina
///elementele cerute, sau daca acestea fac parte deja din alti arbori, voi cauta radacinile
///acestora si le voi uni in arborele curent
///pentru fiecare operatie 2, caut radacinile arborilor care contin elementele cerute
///si verific daca sunt egale
#include <iostream>
#include <fstream>
#include <vector>

using namespace std;

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

vector<int> tata;
int n,m;

int rad(int x)
{
    if(tata[x]==x)
        return x;
    return rad(tata[x]);
}

void reuniune(int x,int y)
{
    if(rad(x)!=rad(y))
        tata[rad(x)]=rad(y);
}
int main()
{
    f>>n>>m;
    tata.resize(n+1, 0);
    for(int i=1;i<=m;i++)
    {
        int op,x,y;
        f>>op>>x>>y;
        if(op==1)
        {
            if(tata[x]==0)
                tata[x]=x;
            if(tata[y]==0)
                tata[y]=x;
            reuniune(x,y);
        }
        else {
            if (rad(x) == rad(y) && rad(x) != 0 || x == y)
                g << "DA\n";
            else
                g << "NU\n";
        }
    }
    return 0;
}