Cod sursa(job #3305505)

Utilizator Iustin_Mircea2010Iustin Mircea Iustin_Mircea2010 Data 2 august 2025 12:08:36
Problema Paduri de multimi disjuncte Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.83 kb
#include <bits/stdc++.h>

using namespace std;

int lead[100005], depth[100005];

int find(int u){
    if(lead[u] == u) return u;
    return lead[u] = find(lead[u]);
}
void unite(int u, int v){
    u = find(u);
    v = find(v);
    if(u == v) return;
    if(depth[u] > depth[v]) swap(u, v);
    lead[u] = v;
    if(depth[u] == depth[v]) depth[v]++;
}

int main(){
    
    ifstream cin("disjoint.in");
    ofstream cout("disjoint.out");
    
    int n, m;
    cin >> n >> m;
    for(int i = 1; i <= n; i++){
        lead[i] = i;
        depth[i] = 1;
    }
    for(int i = 1; i <= m; i++){
        int t, x, y;
        cin >> t >> x >> y;
        if(t == 1){
            unite(x, y);
        }
        else{
            if(find(x) == find(y))
                cout << "DA\n";
            else
                cout << "NO\n";
        }
    }
    return 0;
}