Cod sursa(job #1930023)

Utilizator SagunistuStrimbu Alexandru Sagunistu Data 18 martie 2017 14:05:49
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.2 kb
#include <fstream>
#define nmax 100005

using namespace std;

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

class DisjointSet
{
private:

    void Link(int x,int y)
    {
        if(a[x].r < a[y].r)
            a[x].p = y;
        else
        {
            a[y].p = x;
            if(a[y].r == a[x].r)
                a[x].r ++;
        }
    }

public:

    struct Elem
    {
        int p,r;
    };

    Elem a[nmax];

    void MakeSet(int x)
    {
        a[x].p = x;
        a[x].r = 0;
    }

    int FindSet(int x)
    {
        if(a[x].p != x)
            a[x].p = FindSet(a[x].p);
        return a[x].p;
    }

    void Union(int x,int y)
    {
        Link(FindSet(x),FindSet(y));
    }
};

DisjointSet disjset;

int n,m;

int main()
{
    int t,x,y;
    fin>>n>>m;
    for(int i = 1;i <= n;i++)
        disjset.MakeSet(i);
    for(int i = 0;i < m;i++)
    {
        fin >> t >> x >> y;
        if(t == 1)
            disjset.Union(x,y);
        else
        {
            if(disjset.FindSet(x) == disjset.FindSet(y))
                fout << "DA\n";
            else
                fout << "NU\n";
        }
    }
    return 0;
}