Cod sursa(job #2483022)

Utilizator SergiuS3003Sergiu Stancu Nicolae SergiuS3003 Data 29 octombrie 2019 10:13:27
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.06 kb
#include <iostream>
#include <fstream>
using namespace std;
ifstream f("disjoint.in");
ofstream g("disjoint.out");
const int NMAX = 1000001;
int N, M, T[NMAX], H[NMAX];
int Find(int x)
{
    int y, z;
    for(y = T[x]; y != T[y]; y = T[y]);
    while(x != T[x])
    {
        z = T[x];
        T[x] = y;
        x = z;
    }
    return y;
}
int FindRec(int x)
{
    if(T[x]==x)
        return x;
    return T[x]=FindRec(T[x]);
}
void Union(int x, int y)
{
    if(H[x] < H[y])
        T[x] = y;
    else
    {
        T[y] = x;
        if(H[x] == H[y])
            H[y]++;
    }
}
int main()
{
    int N, M;
    f >> N >> M;
    for(int i = 1; i <= N; i++)
    {
        T[i] = i;
        H[i] = 1;
    }
    while(M--)
    {
        int cod, x, y;
        f >> cod >> x >> y;
        x = FindRec(x);
        y = FindRec(y);
        if(cod == 1)
            Union(x, y);
        else
        {
            if(x == y)
                g << "DA\n";
            else
                g << "NU\n";
        }
    }
    return 0;
}