Cod sursa(job #2806015)

Utilizator AlinaFloreaFlorea Alina AlinaFlorea Data 22 noiembrie 2021 11:36:09
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.23 kb
#include <bits/stdc++.h>

using namespace std;

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

int n, m, cod, x, y, rx, ry;
int t[100005], nod[100005];
int radacina(int x){
    if(x != t[x])
        return t[x] = radacina(t[x]);
    else
        return x;
}
int main()
{
    f >> n >> m;
    ///imbunatatiri: retinem numarul de noduri din fiecare arbore
    ///daca nr de noduri din arbrorele rx este > nr de noduri din arborele ry t[ry] = rx;
    ///arborele mai numeros preia arborele mai putin numeros
    ///daca am nodul x determin radinca rx si mai parcurg inca o data traseul spre radacina si leg
    ///fiecare nod direct de radacina, t[i] = r[x]
    for(int i = 1; i <= n; i++)
        t[i] = i, nod[i] = 1;
    while(m--){
        f >> cod >> x >> y;
        rx = radacina(x);
        ry = radacina(y);
        if(cod == 1){
            if(nod[rx] > nod[ry]){
                nod[rx] += nod[ry];
                t[ry] = rx;
            }
            else{
                nod[ry] += nod[rx];
                t[rx] = ry;
            }
        }
        else{
            if(rx == ry)
                g << "DA\n";
            else
                g << "NU\n";
        }
    }

    return 0;
}