Cod sursa(job #1653516)

Utilizator ZeBuGgErCasapu Andreas ZeBuGgEr Data 16 martie 2016 09:26:42
Problema Paduri de multimi disjuncte Scor 80
Compilator cpp Status done
Runda Arhiva educationala Marime 1.16 kb
#include<cstdio>

int n,m,type,x,y;
int tata[100010];
int sz[100010];

void reuneste(int n1,int n2)
{
    int t1=n1,t2=n2;
    while(tata[t1]!=0)
    {
        t1=tata[t1];
    }
    while(tata[t2]!=0)
    {
        t2=tata[t2];
    }
    if(sz[t1]>sz[t2])
    {
        sz[t1]+=sz[t2];
        tata[n2]=t1;
    }
    else
    {
        sz[t2]+=sz[t1];
        tata[n1]=t2;
    }
}

int exista(int n1,int n2)
{
    while(tata[n1]!=0)
    {
        n1=tata[n1];
    }
    while(tata[n2]!=0)
    {
        n2=tata[n2];
    }
    if(n1==n2)
    {
        return 1;
    }
    return 0;
}

int main()
{
    freopen("disjoint.in","r",stdin);
    freopen("disjoint.out","w",stdout);

    scanf("%d %d",&n,&m);

    for(int i=1;i<=n;i++)
    {
        sz[i]=1;
    }

    for(int i=1;i<=m;i++)
    {
        scanf("%d %d %d",&type,&x,&y);
        if(type==1)
        {
            reuneste(x,y);
        }
        else
        {
            if(exista(x,y)==1)
            {
                printf("DA\n");
            }
            else
            {
                printf("NU\n");
            }
        }
    }
}