Cod sursa(job #2815892)

Utilizator MaraNNedelcu Mara MaraN Data 10 decembrie 2021 16:00:05
Problema Paduri de multimi disjuncte Scor 40
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.83 kb
#include <bits/stdc++.h>

using namespace std;

int sef[100001];

int boss(int a)
{
    if (sef[a] == a)
        return a;
    else
    {
        return sef[a] = boss(sef[a]);
    }
}

void unire(int a, int b)
{
    int boss1 = boss(a);
    int boss2 = boss(b);
    sef[boss2] = boss1;
}

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

    int n, m, cod, x, y, i;
    fin >> n >> m;
    for (i = 1; i <= n; i++)
        sef[i] = i;
    for (i = 1; i <= m; i++)
    {
        fin >> cod >> x >> y;
        if (cod == 1)
        {
            unire(x, y);
        }
        else
        {
            if (boss(x) == boss(y))
                fout << "DA" << endl;
            else fout << "NU" << endl;
        }
    }

    fin.close();
    fout.close();
    return 0;
}