Cod sursa(job #1374575)

Utilizator diana97Diana Ghinea diana97 Data 5 martie 2015 10:07:20
Problema Paduri de multimi disjuncte Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 1.04 kb
#include <iostream>
#include <fstream>

using namespace std;

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

const int NMAX = 100000 + 1;

int n, q;
int tata[NMAX], rang[NMAX];

void initializeaza() {
    for (int i = 1; i <= n; i++) {
        tata[i] = i;
        rang[i] = 1;
    }
}

int radacina(int x) {
    int r = x, aux;
    while (r != tata[r]) r = tata[r];
    while (x != tata[x]) {aux = x; x = tata[x]; tata[aux] = r;}
    return r;
}

inline void uneste(int a, int b) {
    a = radacina(a);
    b = radacina(b);
    if (rang[b] > rang[a]) tata[a] = b;
    else tata[b] = a;
    if (rang[b] == rang[a]) rang[a]++;
}

inline bool sunt_unite(int a, int b) {
    a = radacina(a);
    b = radacina(b);
    return (a == b);
}

void rezolva() {
    int tip, a, b;
    while(q--) {
        f >> tip >> a >> b;
        if (tip == 1) uneste(a, b);
        else {
            if (sunt_unite(a, b)) cout << "DA\n";
            else cout << "NU\n";
        }
    }
}

int main() {
    f >> n >> q;
    initializeaza();
    rezolva();
    return 0;
}