Cod sursa(job #2203049)

Utilizator YouDontNeedMyNameJurcut Paul YouDontNeedMyName Data 10 mai 2018 19:57:13
Problema Paduri de multimi disjuncte Scor 10
Compilator cpp Status done
Runda Arhiva educationala Marime 0.76 kb
#include <bits/stdc++.h>
#define N_MAX 100005
using namespace std;
ifstream in("disjoint.in");
ofstream out("disjoint.out");
int tat[N_MAX],n,m;
int cauta(int a)
{
    int x=a;
    while(tat[a]!=a)
    {
        a=tat[a];
    }
    while(tat[x]!=x)
    {
        int p=tat[x];
        tat[x]=a;
        x=p;
    }
    return a;
}
int main()
{
    in >> n >> m;
    for(int i=1; i<=n; i++)
    {
        tat[i]=i;
    }
    for(int i=0; i<m; i++)
    {
        int x,y,z;
        in >> x >> y >> z;
        if(x==1)
        {
            tat[y]=tat[z];
        }
        else
        {
            if(cauta(y)==cauta(z))
                out << "DA\n";
            else
                out << "NU\n";
        }
    }
    return 0;
}