Cod sursa(job #2697806)

Utilizator alexia208160Popescu Alexia Maria alexia208160 Data 19 ianuarie 2021 18:11:55
Problema Paduri de multimi disjuncte Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.3 kb
#include <fstream>

using namespace std;

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

const int NMAX = 100000;
int v[NMAX + 5], siz[NMAX + 5];

void Find(int x)
{
    if(v[x] == x)
        return ;

    Find(v[x]);
    siz[x] = siz[v[x]];
    v[x] = v[v[x]];
}

bool CommonParent(int x, int y)
{
    if(v[x] == v[y])
        return 1;
    return 0;
}

int main()
{
    int n, m;
    fin >> n >> m;

    for(int i = 1; i <= n; i++)
    {
        v[i] = i;
        siz[i] = 1;
    }

    for(int i = 0; i < m; i++)
    {
        int t, x, y;
        fin >> t >> x >> y;
        if(t == 1)
        {
            if(siz[x] > siz[y])
            {
                Find(x);
                siz[v[x]] += siz[v[y]];
                v[y] = v[x];
                Find(y);
            }
            else
            {
                Find(y);
                siz[v[y]] +=siz[v[x]];
                v[x] = v[y];
                Find(x);
            }
        }
        else
        {
            Find(x);
            Find(y);
            if(CommonParent(x, y))
                fout << "DA\n";
            else
                fout << "NU\n";
        }
    }

    for(int i = 1; i<= n; i++)
        fout << v[i] <<' '<< siz[i] << '\n';
    return 0;
}