Cod sursa(job #2220200)

Utilizator AlexandraCristeaAlexandraCristea AlexandraCristea Data 10 iulie 2018 21:08:02
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 0.92 kb
#include <fstream>

using namespace std;
const int NMAX=100000;
int r[NMAX+5],h[NMAX+5];
int findset(int x) {
    while(r[x]!=x)
        x=r[x];
    return x;
}
void unionset(int x,int y) {
    if(h[x]==h[y])

    {

        h[x]++;
        r[y]=x;
    } else    if(h[x]>h[y])
        r[y]=x;
    else
        r[x]=y;


}
int main() {
    ifstream in("disjoint.in");
    ofstream out("disjoint.out");
    int n,m,i,x,y,rx,ry,cod;
    in >>n>>m;
    for(i=1; i<=n; i++) {
        r[i]=i;
        h[i]=1;
    }
    for(i=1; i<=m; i++) {
        in >>cod>>x>>y;
        rx=findset(x);
        ry=findset(y);
        if(cod==1)
            if(r[x]!=r[y])
                unionset(rx,ry);
        if(cod==2)
                    {if(rx==ry)
                        out <<"DA"<<'\n';
                    else
                        out <<"NU"<<'\n';
                    }
    }
    return 0;
}