Cod sursa(job #1893844)

Utilizator 1475369147896537415369Andrei Udriste 1475369147896537415369 Data 26 februarie 2017 03:30:39
Problema Paduri de multimi disjuncte Scor 50
Compilator cpp Status done
Runda Arhiva educationala Marime 1.31 kb
#include <cstdio>
#include <stack>
using namespace std;

struct node{
    int rank;
    int parent;
}node[100001];

int Find(int target){
    stack<int> path;

    while(node[target].parent != target){
        if(node[node[target].parent].parent != node[target].parent){
            path.push(target);
        }
        target = node[target].parent;
    }
    while(!path.empty()){
        node[path.top()].parent = target;
        path.pop();
    }
    return target;
}
void Union(int x, int y){
    int rootX = Find(x);
    int rootY = Find(y);

    if(node[rootX].rank > node[rootY].rank){
        node[rootY].parent = rootX;
    }else{
        if(node[rootX].rank == node[rootY].rank){
            node[rootY].rank++;
        }
        node[rootX].parent = rootY;
    }
}

int main(){

    int sets, queries, task, x, y;

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

    scanf("%d %d", &sets, &queries);

    for(int i = 1; i <= sets; i++){
        node[i].rank = 1;
        node[i].parent = i;
    }
    for(int i = 1; i <= queries; i++){
        scanf("%d %d %d", &task, &x, &y);

        if(task == 1){
            Union(x, y);
        }else{
            printf("%s\n", (Find(x) == Find(y)) ? "DA" : "NU");
        }
    }
    return 0;
}