Cod sursa(job #2220001)

Utilizator Cristina42Cirstea Cristina-Elena Cristina42 Data 10 iulie 2018 12:38:00
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 0.85 kb
#include <fstream>

using namespace std;
const int nmax=100000;
int r[nmax],h[nmax];
ifstream in("disjoint.in");
ofstream out("disjoint.out");
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() {
    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;
        in>>x>>y;
        rx=findset(x);
        ry=findset(y);
        if(cod==2) {
            if(rx==ry)out<<"DA\n";
            else out<<"NU\n";
        } else {
            if(rx!=ry) {
                 unionset(rx,ry);
            }
        }
    }
    return 0;
}