Cod sursa(job #1005583)

Utilizator mirceadinoMircea Popoveniuc mirceadino Data 5 octombrie 2013 12:20:02
Problema Paduri de multimi disjuncte Scor 10
Compilator cpp Status done
Runda Arhiva educationala Marime 0.92 kb
#include<cstdio>
#define NMAX 100000+5
using namespace std;
int N,M;
int root[NMAX],rang[NMAX];
void unite(int a,int b)
{
    if(rang[a]<rang[b]) //unim in arborele de rang mai mare
        root[a]=b;
    else root[b]=a;
    if(rang[a]==rang[b]) rang[a]++;
}
int find(int a)
{
    int R,b;
    for(R=a;root[R]!=R;R=root[R]); //caut radacina
    for(;root[a]!=a;) //compresia drumurilor
    {
        b=root[a];
        root[a]=R;
        a=b;
    }
    return R;
}
int main()
{
    int i,a,b,t;
    freopen("disjoint.in","r",stdin);
    freopen("disjoint.out","w",stdout);
    scanf("%d%d",&N,&M);
    for(i=1;i<=N;i++)
    {
        root[i]=i; rang[i]=1;
    }
    for(;M;--M)
    {
        scanf("%d%d%d",&t,&a,&b);
        if(t==1)
        {
            unite(a,b);
            continue;
        }
        if(find(a)==find(b)) printf("DA\n");
        else printf("NU\n");
    }
    return 0;
}