Cod sursa(job #2000138)

Utilizator ardutgamerAndrei Bancila ardutgamer Data 12 iulie 2017 18:17:56
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.06 kb
#include <cstdio>

using namespace std;
const int NMAX =100005;
int n;
int t[NMAX] , h[NMAX];

int FindSet(int x)
{
    while(t[x] != x)
        x = t[x];
    return x;
}

void UnionSet(int x,int y)
{
    if(h[x] == h[y])
    {
        t[y] = x;
        h[x]++;
    }
    else if(h[x] > h[y])
        t[y] = x;
    else
        t[x] = y;
}
int main()
{
    freopen("disjoint.in","r",stdin);
    freopen("disjoint.out","w",stdout);
    int m , cod , x , y ;
    scanf("%d%d",&n,&m);
    for(int i = 1 ; i <= n ; i++)
    {
        t[i] = i;
        h[i] = 1;
    }
    for(int i = 1 ; i <= m ; i++)
    {
        scanf("%d%d%d",&cod,&x,&y);
        if(cod == 1)
        {
            x = FindSet(x);
            y = FindSet(y);
            if(x != y)
            UnionSet(x,y);
        }
        else
        {
            int ans1 = FindSet(x);
            int ans2 = FindSet(y);
            if(ans1 == ans2)
                printf("DA\n");
            else
                printf("NU\n");
        }
    }
    return 0;
}