Cod sursa(job #3041067)

Utilizator stefan05Vasilache Stefan stefan05 Data 30 martie 2023 21:49:53
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.91 kb
#include <fstream>
#include <vector>

#define NMAX 100005

using namespace std;

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

int n, m;
int cod, x, y;
int t[NMAX];
int rang[NMAX];
int i, j;

int radacina(int);
void unire(int, int);

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

    for (i = 1; i <= m; ++i)
    {
        fin >>cod>>x>>y;
        if (cod == 1) unire(x, y);
        else
            fout <<(radacina(x) == radacina(y)? "DA": "NU")<<'\n';
    }
    fout.close();
    return 0;
}

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

void unire(int x, int y)
{
    int r1 = radacina(x), r2 = radacina(y);
    if (r1 != r2)
        if (rang[r1] < rang[r2])
            t[r1] = r2;
        else
            {
                t[r2] = r1;
                if (rang[r1] == rang[r2]) rang[r1] ++;
            }
}