Cod sursa(job #1462127)

Utilizator Burbon13Burbon13 Burbon13 Data 17 iulie 2015 10:21:02
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.07 kb
#include <cstdio>
using namespace std;

const int nmx = 100005;

int n, m;
int parinte[nmx], rang[nmx];

int gaseste(int x){
    int f = parinte[x];

    while(parinte[f] != f)
        f = parinte[f];

    while(parinte[x] != x){
        int aux = parinte[x];
        parinte[x] = f;
        x = aux;
    }

    return f;
}

void uneste(int x1, int x2){
    if(rang[x1] > rang[x2])
        parinte[x2] = x1;
    else
        parinte[x1] = x2;
    if(rang[x1] == rang[x2])
        ++ rang[x2];
}

int main() {
    freopen("disjoint.in", "r", stdin);
    freopen("disjoint.out", "w", stdout);

    scanf("%d %d", &n, &m);

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

    for(int i = 1; i <= m; ++i) {
        int c, n1, n2;
        scanf("%d %d %d", &c, &n1, &n2);

        if(c == 2) {
            if(gaseste(n1) == gaseste(n2))
                printf("DA\n");
            else
                printf("NU\n");
        } else
            uneste(gaseste(n1),gaseste(n2));
    }

    return 0;
}