Cod sursa(job #3218490)

Utilizator ililogIlinca ililog Data 27 martie 2024 12:00:17
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.48 kb
#include <iostream>
#include<fstream>
#define NMAX 100001
using namespace std;

ifstream fin("disjoint.in");
ofstream fout("disjoint.out");

int tata[NMAX];
int n, m;
int h[NMAX]; ///inaltimea
int op, a, b;


int Find(int x) {
    int r, y;
    ///mai intai aflu radacina arborelui din care face parte x
    r = x;
    while (tata[r] != r) {
        r = tata[r];
    }
    ///parcurg din nou drumul de la x la r si fac toate nodurile fii ai lui r

    while(tata[x]!=r)
    {
         /// de la x la radacina toti parintii nodurilor = radacina
        int aux=tata[x];
        tata[x]=r;
        x=aux;
    }

    return r;
}

void Union(int x, int y)
{
    int rx=Find(x);
    int ry=Find(y);
    if(rx!=ry)
    {
        if(h[rx]<h[ry]) ///unim arborele mic la cel mare rx la ry
                tata[rx]=ry;
        else
        {
            if(h[ry]<h[rx]) ///ry la rx
                tata[ry]=rx;
            else ///sunt egale
            {
                tata[rx]=ry;
                h[ry]++;
            }

        }
    }
}

int main()
{
    fin >> n >> m;
    for (int i = 1; i<=n; i++) {
        h[i] = 1, tata[i] = i;
    }

    for (int i = 1; i<=m; i++) {
        fin >> op >> a >> b;
        if (op == 1) {
            Union(a, b);
        } else {
            if (Find(a) == Find(b)) {
                fout << "DA\n";
            } else {
                fout << "NU\n";
            }
        }
    }
    return 0;
}