Cod sursa(job #3218538)

Utilizator Maia_TomitaMaia Tomita Maia_Tomita Data 27 martie 2024 12:39:22
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.32 kb
#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 cod, a, b;


int Find(int x) {
    int r, y;
    ///mai intai aflu rad 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) {
        y = tata[x];
        tata[x]=r;
        x=y;
    }

    return r;
}

void Union(int x, int y)
{
    int rx, ry;
    rx = Find(x);
    ry = Find(y);

    if (rx != ry)
        {
        if (h[rx]<h[ry]) {
            tata[rx]=ry;
            }
        else if (h[ry]<h[rx])
            {
            tata[ry]=rx;
            }
        else
            {
            tata[ry]=rx, h[ry]++;
            }
    }
}


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

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