Cod sursa(job #899255)

Utilizator anca1243Popescu Anca anca1243 Data 28 februarie 2013 13:35:56
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 0.91 kb
#include <iostream>
#include <fstream>
using namespace std;
ifstream in("disjoint.in");
ofstream out("disjoint.out");
const int N=100001;
int n,m,set[N],h[N],k;
void init()
{
    for(int i=1;i<=n;i++)
    {
        set[i]=i;
        h[i]=1;
    }
}
void reuniune(int a,int b)
{
    if(h[a]==h[b])
    {
        h[a]=h[a]+1;
        set[b]=a;
    }
    else
        if(h[a]>h[b])
            set[b]=a;
        else
            set[a]=b;
}
int ver(int x)
{
    int r=x;
    while(set[r]!=r)
        r=set[r];
    int i=x,j;
    while(i!=r)
    {
        j=set[i];
        set[i]=r;
        i=j;
    }
    return r;
}
int main()
{
    int x,y;
    in>>n>>m;
    init();
    for(int i=1;i<=m;i++)
    {
        in>>k>>x>>y;
        if(k==1)
            reuniune(ver(x),ver(y));
        else
            if(ver(x)==ver(y))
                out<<"DA"<<'\n';
            else
                out<<"NU"<<'\n';

    }
    return 0;
}