Cod sursa(job #2567659)

Utilizator TudorCristeaCristea Tudor TudorCristea Data 3 martie 2020 18:16:38
Problema Paduri de multimi disjuncte Scor 30
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.21 kb
#include <iostream>
#include <fstream>
using namespace std;
ifstream fin("disjoint.in");
ofstream fout("disjoint.out");

int N,M;

int P[505],GR[505];

int parinte (int nod)
{
    while (P[nod]!=nod)
    {
        nod=P[nod];
    }
    return nod;
}

void reuniune (int nod1,int nod2)
{
    int p_nod1=parinte(nod1);
    int p_nod2=parinte(nod2);
    if (p_nod1!=p_nod2)
    {
        if (GR[p_nod1]<GR[p_nod2])
        {
            P[p_nod1]=p_nod2;
        }
        else if (GR[p_nod1]>GR[p_nod2])
        {
            P[p_nod2]=p_nod1;
        }
        else
        {
            P[p_nod1]=p_nod2;
            ++GR[p_nod2];
        }
    }
}

int main()
{
    fin >> N >> M;
    int i;
    for (i=1;i<=N;++i)
    {
        P[i]=i;
        GR[i]=1;
    }
    for (i=1;i<=M;++i)
    {
        int a,b,t;
        fin >> t >> a >> b;
        if (t==1)
        {
            reuniune(a,b);
        }
        else
        {
            if (parinte(a)==parinte(b))
            {
                fout << "DA\n";
            }
            else
            {
                fout << "NU\n";
            }
        }
    }
    fin.close();
    fout.close();
    return 0;
}