Cod sursa(job #2806013)

Utilizator Edyci123Bicu Codrut Eduard Edyci123 Data 22 noiembrie 2021 11:35:53
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.14 kb
#include <bits/stdc++.h>
#define DIM 100005

using namespace std;

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

int n, m, t[DIM], nrNod[DIM];

void init()
{
    for(int i = 1; i <= n; i++)
        t[i] = i, nrNod[i] = 1;
}

int radacina(int x)
{
    if(x != t[x])
        return t[x] = radacina(t[x]);
    else
        return x;
}

int main()
{
    f >> n >> m;

    init();
    for(int i = 1; i <= m; i++)
    {
        int cod, x, y;
        f >> cod >> x >> y;

        int radx = radacina(x);
        int rady = radacina(y);

        if(cod == 1)
        {
            if(radx != rady)
            {
                if(nrNod[radx] > nrNod[rady])
                {
                    nrNod[radx] += nrNod[rady];
                    t[rady] = radx;
                }
                else
                {
                    nrNod[rady] += nrNod[radx];
                    t[radx] = rady;
                }
            }
        }
        else
        {
            if(radx == rady)
                g << "DA\n";
            else
                g << "NU\n";
        }
    }

    return 0;
}