Cod sursa(job #2280130)

Utilizator aturcsaTurcsa Alexandru aturcsa Data 10 noiembrie 2018 11:39:27
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.79 kb
#include <iostream>
#include <fstream>
using namespace std;
ifstream fin("disjoint.in");
ofstream fout("disjoint.out");
int P[100001];
int n,m;
int radacina(int nod)
{
    while(P[nod]!=nod)
        nod=P[nod];
    return nod;
}
int q(int nod)
{
    if(P[nod]!=nod)
        P[nod]=q(P[nod]);
    return P[nod];
}
int main()
{
    fin>>n>>m;
    for(int i=1;i<=n;i++)
        P[i]=i;
    for(int i=1;i<=m;i++)
    {
        int cod,x,y;
        fin>>cod>>x>>y;
        if(cod==1)
            {
                P[radacina(x)]=radacina(y);
            }

        else
        {
            P[x]=q(x);
            P[y]=q(y);
            if(P[x]==P[y])
                fout<<"DA\n";
            else
                fout<<"NU\n";

        }
    }
    return 0;
}