Cod sursa(job #2346589)

Utilizator radurotaruRotaru Radu Stefan radurotaru Data 17 februarie 2019 20:37:04
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.96 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++;
        }
    }
}
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;
        if(x==1)
            Union(a,b);
        else
        {
            if(find(a)==find(b))
                g<<"DA"<<'\n';
            else
                g<<"NU"<<'\n';
        }
    }
    return 0;
}