Cod sursa(job #3138079)

Utilizator petric_mariaPetric Maria petric_maria Data 17 iunie 2023 12:46:33
Problema Paduri de multimi disjuncte Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.7 kb
#include <bits/stdc++.h>
using namespace std;
ifstream f("disjoint.in");
ofstream g("disjoint.out");
int n,m,t[100001],v[100001];
int rad(int k){
    int x=k,y;
    while(t[k]) k=t[k];
    while(x!=k){
        y=x; x=t[x]; t[y]=k;
    }
    return k;
}
int uni(int k,int p){
    if(v[k]==v[p]) v[k]++;
    if(v[k]>v[p]) t[p]=k;
    else t[k]=p;
}
int main()
{
    int i,c,x,y;
    f>>n>>m;
    for(i=1;i<=n;++i) v[i]=1,t[i]=0;
    for(i=1;i<=m;++i){
        f>>c>>x>>y;
        if(c==1){
            x=rad(x); y=rad(y);
            if(x!=y) uni(x,y);
        }
        else{
            if(rad(x)==rad(y)) g<<"DA\n";
            else g<<"NU\n";
        }
    }
    return 0;
}