Cod sursa(job #2075152)

Utilizator MDiana15Diana M MDiana15 Data 25 noiembrie 2017 11:34:05
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 0.82 kb
#include <fstream>

using namespace std;
ifstream f("disjoint.in");
ofstream g("disjoint.out");
int n,m,i,j,t[100001],h[100001],o,a,b,r1,r2;
int findd(int x)
{
    int r=x;
    while(t[r])r=t[r];
    int y=x;;
    while(y!=r)
    {
        int t1=t[y];
        t[y]=r;
        y=t1;
    }
    return r;
}
void unionn(int x,int y)
{
    int c;
    if(h[x]<h[y]){t[x]=y;c=y;}
    else {t[y]=x;c=x;}
    if(h[x]==h[i]){h[c]++;}
}

int main()
{
    f>>n>>m;
    for(i=1;i<=n;i++)
    {
    t[i]=0;
    h[i]=1;
    }
    for(i=1;i<=m;i++)
    {
        f>>o>>a>>b;
        r1=findd(a);
        r2=findd(b);
        if(o==2)
        {
            if(r1==r2)g<<"DA"<<'\n';
            else g<<"NU"<<'\n';
        }
        if(o==1)
        { unionn(r1,r2);

        }
    }
    return 0;
}