Cod sursa(job #3197074)

Utilizator vozian.anghelinaAnghelina Vozian vozian.anghelina Data 25 ianuarie 2024 16:46:35
Problema Paduri de multimi disjuncte Scor 70
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.75 kb
#include <bits/stdc++.h>
using namespace std;
int n, m, T[100100];

int find(int t){
    if(T[t] == t){
        return T[t];
    }

    return find(T[t]);
}

void unit(int a, int b){
    if(T[a] == a && T[b] == b){
        T[a] = T[b];
    } else if(T[b] == b){
        unit(T[a], b);
        T[a] = T[b];
    } else {
        unit(a, T[b]);
        T[a] = T[b];
    }
}

int main(){
    ifstream cin("disjoint.in");
    ofstream cout("disjoint.out");
    cin >> n >> m;
    for(int i=1; i<=n; i++){
        T[i] = i;
    }

    int k, a, b;
    for(int i=1; i<=m; i++){
        cin >> k >> a >> b;
        if(k==1 && find(a) != find(b)){
            unit(a, b);
        } else {
            cout << (find(T[a]) == find(T[b]) ? "DA" : "NU") << endl;
        }
    }
}