Cod sursa(job #2346581)

Utilizator radurotaruRotaru Radu Stefan radurotaru Data 17 februarie 2019 20:32:08
Problema Paduri de multimi disjuncte Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.97 kb
#include <fstream>

using namespace std;

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

struct
{
    int parent, rank;
} p[100001];
int n,m,x,a,b;
int find(int x)
{
    if(p[x].parent !=x)
        p[x].parent=find(p[x].parent);
    return p[x].parent;
}

int Union(int x, int y)
{
    int q=find(x);
    int w=find(y);
    if(q!=w)
    {
        if(p[q].rank > p[w].rank)
            p[w].parent=q;
        if(p[w].rank > p[q].rank)
            p[q].parent=w;
        if(p[w].rank == p[q].rank)
        {
            p[q].parent=w;
            p[w].rank++;
        }
        return 0;
    }
    return 1;
}
int main()
{
    f>>n>>m;
    for(int i=1; i<=n; i++)
        p[i].parent=i;
    for(int i=1; i<=m; i++)
    {
        f>>x>>a>>b;
        int r = Union(a,b);
        if(x==2)
        {
            if(r)
                g<<"DA"<<'\n';
            else
                g<<"NU"<<'\n';
        }
    }
    return 0;
}