Cod sursa(job #2337502)

Utilizator BotzkiBotzki Botzki Data 6 februarie 2019 14:46:06
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.96 kb
#include <fstream>

using namespace std;
ifstream fin("disjoint.in");
ofstream fout("disjoint.out");
const int NMAX=100000;
int t[NMAX+5], h[NMAX+5];
int FindSet(int x)
{
    int cx=x, aux;
    while(x!=t[x])
        x=t[x];
     //Optimizez radacina pt urmatoarele operatii
     while(cx!=x)
     {
         aux=t[cx];
         t[cx]=x;
         cx=aux;
     }
    return x;
}
void UnionSet(int x, int y)
{
    //x si y sunt radacinile celor 2 arbori
    t[x]=y;
}
int main()
{
    int i, n, k, a, b, x, ta, tb;
    fin>>n>>k;
    for(i=1;i<=n;i++)
        t[i]=i;
    for(i=1;i<=k;i++)
    {
        fin>>x>>a>>b;
        if(x==1)
        {
           ta=FindSet(a);
           tb=FindSet(b);
           UnionSet(ta, tb);
        }
        else
        {
          ta=FindSet(a);
          tb=FindSet(b);
          if(ta==tb)
            fout<<"DA"<<"\n";
          else
            fout<<"NU"<<"\n";
        }
    }
    return 0;
}