Cod sursa(job #2026997)

Utilizator circeanubogdanCirceanu Bogdan circeanubogdan Data 25 septembrie 2017 14:27:52
Problema Paduri de multimi disjuncte Scor 40
Compilator cpp Status done
Runda Arhiva educationala Marime 0.91 kb
#include <fstream>

using namespace std;

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

int n, m, t[100002], rx, ry, val, x, y;

int rad(int x)
{
    while(t[x] > 0)
        x = t[x];
    return x;
}

int main()
{
    f>>n>>m;
    for(int i = 1; i <= n; ++ i)
        t[i] = -1;
    for(int i = 1; i <= m ; ++ i)
    {
        f>>val>>x>>y;
        rx = rad(x);
        ry = rad(y);
        if(val == 2)
        {
            if(rx == ry) g<<"DA"<<'\n';
            else g<<"NU"<<'\n';
        }
        else
        {
            if(rx != ry)
            {
                if(t[rx] < t[ry])
                {
                    t[rx] += t[ry];
                    t[ry] = rx;
                }
                else
                {
                    t[ry] += t[ry];
                    t[rx] = ry;
                }
            }
        }
    }
    return 0;
}