Cod sursa(job #3306355)

Utilizator vlad7654vladimir manescu vlad7654 Data 9 august 2025 20:51:35
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.95 kb
#include <bits/stdc++.h>
using namespace std;
ifstream fin("disjoint.in");
ofstream fout("disjoint.out");
const int NMAX=1e5;
vector<int> parent(NMAX+5), status(NMAX+5);
int find_set(int nod){
    if(nod==parent[nod]){
        return nod;
    }
    return parent[nod]=find_set(parent[nod]);
}
void union_set(int x, int y) {
    x=find_set(x);
    y=find_set(y);
    if(x!=y) {
        if(status[x]<status[y]){
            swap(x, y);
        }
        parent[y]=x;
        status[x]+=status[y];
    }
}
int main(){
    int n, m;
    fin>>n>>m;
    for(int i=1;i<=n;i++){
        parent[i]=i;
        status[i]=1;
    }
    for(int j=1;j<=m;j++){
        int tip, x, y;
        fin>>tip>>x>>y;
        if(tip==1){
            union_set(x,y);
        }else if(tip==2){
            if(find_set(x)==find_set(y)){
                fout<<"DA"<<'\n';
            }else{
                fout<<"NU"<<'\n';
            }
        }
    }

}