Cod sursa(job #1887908)

Utilizator danyvsDan Castan danyvs Data 21 februarie 2017 20:20:37
Problema Paduri de multimi disjuncte Scor 70
Compilator cpp Status done
Runda Arhiva educationala Marime 0.91 kb
#include <fstream>
#include <cstring>

using namespace std;

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

const int NMAX = 100005;

int n, m;
int par[NMAX], ht[NMAX];

int root(int node)
{
    while (par[node])
        node = par[node];
    return node;
}

void unite(int a, int b)
{
    if (ht[a] > ht[b])
        par[b] = a;
    else
        {
         par[a] = b;
         if (ht[a] == ht[b])
            ++ ht[a];
        }
}

int main()
{
    fin >> n >> m;
    memset(ht, 1, sizeof(ht));
    for (int i = 0; i < m; ++ i)
        {
         int t, x, y;
         fin >> t >> x >> y;
         int a = root(x);
         int b = root(y);
         if (t == 1)
            unite(a, b);
         else
            if (a == b)
                fout << "DA\n";
            else
                fout << "NU\n";
        }
    fin.close();
    fout.close();
    return 0;
}