Cod sursa(job #2943176)

Utilizator Diana_14Diana Muscalu Diana_14 Data 20 noiembrie 2022 17:41:11
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.58 kb
#include <iostream>
#include <fstream>
#include <vector>

using namespace std;

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

vector <int>tata;
vector <int>inaltime;

int parinte(int x)
{
    ///e radacina
    if(tata[x] == 0)
        return x;

    ///repetam pana ajungem la radacina
    return tata[x] = parinte(tata[x]);
}

int main()
{
    int n, m, cod, x, y;
    f>>n>>m;
    tata.resize(n+1, 0);
    inaltime.resize(n+1, 0);

    for(int i = 0; i < m; i++)
    {
        f>>cod>>x>>y;

        if(cod == 1)
        {
            ///cautam radicinile nodurilor
            int rad_x = parinte(x);
            int rad_y = parinte(y);

            ///verificam care ramura acopera mai multe noduri si in functie de asta setam parintele nodurilor curente
            /// pe primul caz daca ramura lui x acopera mai multe atunci cand ne legam nodurile il punem pe x ca fiind parintele radacinii lui y
            if(inaltime[x] > inaltime[y])
                tata[rad_y] = rad_x;
            else
            {
                ///daca nu facem invers
                tata[rad_x] = rad_y;

                ///iar daca ramurile sunt egale
                //if(inaltime[rad_x] == inaltime[rad_y])
                  //  inaltime[rad_y] += 1;
            }
        }
        else
        {
            ///verificam daca au aceeasi radacina
            int rad_x = parinte(x);
            int rad_y = parinte(y);
            if(rad_x == rad_y)
                g<<"DA\n";
            else
                g<<"NU\n";
        }
    }

    return 0;
}