Cod sursa(job #2419675)

Utilizator antonia.avadaneiAvadanei Antonia antonia.avadanei Data 9 mai 2019 10:52:27
Problema Paduri de multimi disjuncte Scor 40
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.14 kb
#include <iostream>
#include <fstream>
//grad[x] reprez nr de noduri din comp
//find father
int tata[100000],grad[100000];
using namespace std;
int find_father(int nod)
{
    if(tata[nod]==nod)
        return nod;
    tata[nod]=find_father(tata[nod]);
    return tata[nod];
}
int main()
{
    ifstream in("disjoint.in");
    ofstream out("disjoint.out");
    int n,m,i,x,y,cod;
    in>>n>>m;
    for(i=1;i<=n;i++)
    {
        tata[i]=i;
        grad[i]=1;
    }
    for(i=1;i<=m;i++)
    {
        in>>cod>>x>>y;
        if(cod==1)
        {
            int fx=find_father(x);
            int fy=find_father(y);
            if(grad[fx]<grad[fy])
            {
                tata[fx]=fy;
                grad[fy]+=grad[fx];
            }
            else
            {
                tata[fy]=fx;
                grad[fx]+=grad[fy];
            }

        }
        else
            if(cod==2)
        {
            int fx=find_father(x);
            int fy=find_father(y);
            if(fx==fy)
                out<<"DA"<<endl;
            else
                out<<"NU"<<endl;
        }
    }
    return 0;
}