Cod sursa(job #1969467)

Utilizator caprariuapCaprariu Alin-Paul caprariuap Data 18 aprilie 2017 14:40:58
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 0.81 kb
#include <fstream>
#include <cstring>
#include <vector>
#include <algorithm>
#include <stack>
#include <iomanip>
#include <queue>
using namespace std;

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

#define nmax 100010
int p[nmax],n,m,i;

int Find(int x)
{
    int rad=x;
    while (rad!=p[rad])
        rad=p[rad];
    while (x!=p[x])
    {
        int aux=x;
        x=p[x];
        p[aux]=rad;
    }
    return rad;
}

void Union(int x, int y)
{
    p[x]=y;
}

int main()
{
    fin >> n >> m;
    for (i=1; i<=n; i++)
        p[i]=i;
    for (i=1; i<=m; i++)
    {
        int op,x,y;
        fin >> op >> x >> y;
        x=Find(x);
        y=Find(y);
        if (op==2)
            fout << ( x==y ? "DA\n" : "NU\n");
        else
            Union(x,y);
    }
}