Cod sursa(job #2915256)

Utilizator bogdan14042005Dobrinoiu Bogdan bogdan14042005 Data 22 iulie 2022 10:21:26
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.85 kb
#include <fstream>
using namespace std;
ifstream fin ("disjoint.in");
ofstream fout ("disjoint.out");
const int NMAX=100000;
int t[NMAX+5], h[NMAX+5];
int FindRoot(int x)
{
    while(x!=t[x])
        x=t[x];
    return x;
}
void UnionSet(int x, int y)
{
    if(h[x]>h[y])
        t[y]=x;
    else if(h[y]>h[x])
        t[x]=y;
    else{
        t[y]=x;
        h[x]++;
    }
}
int main(){
    int n,m,cer,x,y,tx,ty,i;
    fin>>n>>m;
    for(i=1;i<=n;i++)
    {
        t[i]=i;
        h[i]++;
    }
    for(i=1;i<=m;++i)
    {
        fin>>cer>>x>>y;
        tx=FindRoot(x);
        ty=FindRoot(y);
        if(cer==1&&tx!=ty)
            UnionSet(tx, ty);
        else
        {
            if(ty==tx)
                fout<<"DA"<<"\n";
            else
                fout<<"NU"<<"\n";
        }
    }
    return 0;
}