Cod sursa(job #2140671)

Utilizator eduardvintilaVintila Eduard eduardvintila Data 23 februarie 2018 19:20:12
Problema Paduri de multimi disjuncte Scor 40
Compilator cpp Status done
Runda Arhiva educationala Marime 1 kb
#include <iostream>
#include <fstream>

using namespace std;

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

struct multime
{
    int tata,rang=0;
}p[100005];

int root(int i)
{
    if (p[i].tata==i)
        return i;
    else
    {
        p[i].tata=root(p[i].tata);
        return p[i].tata;
    }
}
void Union(int x, int y)
{
    x=root(x);
    y=root(y);
    if (x!=y)
    {
        if (p[x].rang<p[y].rang)
            p[x].tata=y;
        else if (p[y].rang<p[x].rang)
            p[y].tata=x;
        else
        {
            p[y].tata=x;
            p[x].rang++;
        }
    }
}

int main()
{
    int n,m,i,x,y,cod;
    f>>n>>m;
    for (i=1;i<=n;i++)
        p[i].tata=i;
    for (i=1;i<=m;i++)
    {
        f>>cod>>x>>y;
        if (cod==1)
        {
            Union(x,y);
        }
        else
        {
            if (root(x)==root(y))
                g<<"DA"<<endl;
            else g<<"NU"<<endl;
        }
    }
    return 0;
}