Cod sursa(job #2203045)

Utilizator YouDontNeedMyNameJurcut Paul YouDontNeedMyName Data 10 mai 2018 19:48:16
Problema Paduri de multimi disjuncte Scor 10
Compilator cpp Status done
Runda Arhiva educationala Marime 0.92 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]!=-1)
    {
        a=tat[a];
    }
    while(tat[x]!=-1)
    {
        int p=tat[x];
        tat[x]=a;
        x=p;
    }
    return a;
}
void unite(int a, int b)
{
    tat[a]=b;
}
int main()
{
    in >> n >> m;
    for(int i=1; i<=n; i++)
    {
        tat[i]=-1;
    }
    for(int i=0; i<m; i++)
    {
        int x,y,z;
        in >> x >> y >> z;
        if(x==1)
        {
            if(cauta(y)!=cauta(z))
            {
                unite(y,z);
            }
            else
                return 0;
        }
        else
        {
            if(cauta(y)==cauta(z))
                out << "DA\n";
            else
                out << "NU\n";
        }
    }
    return 0;
}