Cod sursa(job #2833922)

Utilizator SabailaCalinSabaila Calin SabailaCalin Data 15 ianuarie 2022 23:37:49
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.41 kb
#include <iostream>
#include <fstream>

using namespace std;

ifstream f ("disjoint.in");
ofstream g ("disjoint.out");

int n, m;
int parent[100001];
int rang[100001];

int Find(int x)
{
    if (x != parent[x])
    {
        parent[x] = Find(parent[x]);
    }
    return parent[x];
}

void Union(int x, int y)
{
    int rootX = Find(x);
    int rootY = Find(y);
    if (rootX != rootY)
    {
        if (rang[rootX] > rang[rootY])
        {
            parent[rootY] = rootX;
        }
        else
        {
            parent[rootX] = rootY;
            if (rang[rootX] == rang[rootY])
            {
                rang[rootY]++;
            }
        }
    }
}

int main()
{
    f >> m >> n;

    for (int i = 1; i <= m; i++)
    {
        parent[i] = i;
    }

    for (int i = 1; i <= n; i++)
    {
        int option, x, y;
        f >> option;

        switch (option)
        {
            case 1:
                f >> x >> y;
                Union(x, y);
                break;
            case 2:
                f >> x >> y;
                if (Find(x) == Find(y))
                {
                    g << "DA\n";
                }
                else
                {
                    g << "NU\n";
                }
                break;
        }
    }

//    for (int i = 1; i <= m; i++)
//    {
//        cout << parent[i] << " ";
//    }
}