Cod sursa(job #1857129)

Utilizator GeorginskyGeorge Georginsky Data 25 ianuarie 2017 20:42:19
Problema Paduri de multimi disjuncte Scor 40
Compilator cpp Status done
Runda Arhiva educationala Marime 0.79 kb
#include <iostream>
#include <cstdio>
#include <unordered_map>
using namespace std;
struct Element{int r, p;};
unordered_map<int,Element> u;
int n, m;
void make_set(int val){
    u[val].p=-1;
    u[val].r=0;
}

int find_set(int val){
    if(u[val].p==-1)return val;
    int x=find_set(u[val].p);
    u[val].p=x;
    return x;
}

void merge_sets(int x, int y){
    int set1=find_set(x);
    int set2=find_set(y);
    u[set2].p=set1;
}

int main(){
    freopen("disjoint.in", "r", stdin);
    freopen("disjoint.out", "w", stdout);
    cin>>n>>m;
    int op, a, b;
    for(int i=1; i<=n; i++)make_set(i);
    for(int i=1; i<=m; i++){
        cin>>op>>a>>b;
        if(op==1)merge_sets(a, b);
        else cout<<((find_set(a)==find_set(b))?"DA":"NU")<<"\n";
    }
    return 0;
}