Cod sursa(job #2912989)

Utilizator lifusorin.13Lifu Sorin lifusorin.13 Data 12 iulie 2022 09:40:23
Problema Paduri de multimi disjuncte Scor 10
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.89 kb
#include <bits/stdc++.h>

using namespace std;
ifstream fin("disjoint.in");
ofstream fout("disjoint.out");
int x, y, cod, n, m, tt[100001], sz[100001];

int Find(int nod)
{
    if (tt[nod] == nod)
        return nod;
    return tt[nod] = Find(tt[nod]);
}

void Union(int a, int b)
{
    if (a == b)
        return;
    if (sz[a] >= sz[b]) {
        sz[a] += sz[b];
        tt[b] = a;
    }
    else {
        sz[b] += sz[a];
        tt[a] = b;
    }
}

int main()
{
    fin >> n >> m;
    for (int i = 1; i <= n; i++) {
        tt[i] = i;
        sz[i] = 1;
    }
    for (int i = 1; i <= m; i++) {
        fin >> cod >> x >> y;
        if (cod == 2) {
            if(tt[x] == tt[y])
                fout << "DA\n";
            else
                fout << "NU\n";
        }
        else if (cod == 1) {
            Union(x, y);
        }
    }
    return 0;
}