Cod sursa(job #2075143)

Utilizator darisavuSavu Daria darisavu Data 25 noiembrie 2017 11:30:44
Problema Paduri de multimi disjuncte Scor 10
Compilator cpp Status done
Runda Arhiva educationala Marime 0.79 kb
#include <fstream>

using namespace std;
ifstream f("disjoint.in");
ofstream g("disjoint.out");
int n,m,i,j,t[100005],c,a,b,h[100005];
void unionn(int x,int y)
{
    int c1;
    if(h[x]<h[y]){t[x]=y;c1=y;}
    else {t[y]=x;c1=x;}
    if(h[x]==h[y]) h[c1]++;
}
int findd(int x)
{
    int r=x;
    while(t[r]) r=t[r];
    int y=x;
    while(t[y])
    {
        int t1=t[y];
        t[y]=r;
        y=t1;
    }
    return r;
}
int main()
{
    int r1,r2;
    f>>n>>m;
    for(i=1;i<=n;i++) {t[i]=0;h[i]=1;}
    for(i=1;i<=m;i++)
    {
        f>>c>>a>>b;
        if(c==2)
        {
            r1=findd(a);
            r2=findd(b);
            if(r1==r2) g<<"DA";
            else g<<"NU";
            g<<'\n';
        }
        else unionn(a,b);

    }
    return 0;
}