Cod sursa(job #1189555)

Utilizator flore77Simion Florentin flore77 Data 23 mai 2014 10:10:36
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.15 kb
#include <iostream>
#include <fstream>
using namespace std;

#define NMAX 100020

int parent[NMAX];
int size[NMAX];

int Find (int x) {
    int x_copy = x;
    while (parent[x] > 0)
        x = parent[x];
    while (parent[x_copy] > 0) {
        int x_prec = x_copy;
        x_copy = parent[x_copy];
        parent[x_prec] = x;
    }
    return x;
}

void Union (int x, int y) {
    int rx = Find(x);
    int ry = Find(y);
    if (size[parent[rx]] > size[parent[ry]]) {
        parent[ry] = rx;
        size[rx] += size[ry];
    }
    else {
        parent[rx] = ry;
        size[ry] += size[rx];
    }
}

int main() {
    ifstream in;
    in.open("disjoint.in");
    ofstream out;
    out.open("disjoint.out");
    int n, m, cod, nr1, nr2, i;
    in >> n >> m;
    for (i = 1; i <= n; i++)
        size[i] = 1;
    for (i = 0; i < m; i++) {
        in >> cod >> nr1 >> nr2;
        if (cod == 1)
            Union(nr1,nr2);
        else {
            if (Find(nr1) == Find(nr2))
                out << "DA\n";
            else
                out << "NU\n";
        }
    }
    in.close();
    out.close();
    return 0;
}