Cod sursa(job #2162952)

Utilizator FunnyStockyMihnea Andreescu FunnyStocky Data 12 martie 2018 16:05:52
Problema Paduri de multimi disjuncte Scor 70
Compilator cpp Status done
Runda Arhiva educationala Marime 0.86 kb
#include <fstream>

using namespace std;

ifstream fin("disjoint.in");
ofstream fout("disjoint.out");

const int nmax=100000;
int n,m;

int h[nmax+5];
int t[nmax+5];

int FindSet(int nod)
{
    while(t[nod]!=nod)
        nod=t[nod];
    return nod;
}
void UniteSet(int x,int y)
{
    if(h[x]==h[y])
    {
        t[x]=y;
        h[x]++;
    }
    else
    {
        if(h[x]>h[y])
            t[y]=x;
        else
            t[x]=y;
    }
}
int main()
{
    for(int i=1;i<=nmax;i++)
    {
        h[i]=1;
        t[i]=i;
    }
    fin>>n>>m;
    while(m--)
    {
        int tip,x,y;
        fin>>tip>>x>>y;
        if(tip==1)
            UniteSet(FindSet(x),FindSet(y));
        else
            if(FindSet(x)==FindSet(y))
                fout<<"DA\n";
            else
                fout<<"NU\n";
    }
    return 0;
}