Cod sursa(job #2806320)

Utilizator StefaniaCriStefania Cristea StefaniaCri Data 22 noiembrie 2021 15:28:42
Problema Paduri de multimi disjuncte Scor 70
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.99 kb
#include <iostream>
#include <fstream>
using namespace std;
#define Nmax 100007
ifstream f("disjoint.in");
ofstream g("disjoint.out");

int parinte[Nmax],greutate[Nmax];
int find(int x)
{
    if(parinte[x]==x) return x;
    return find(parinte[x]);

}

void unite(int x, int y)
{
    int parinte_x = find(x);
    int parinte_y = find(y);
    if(greutate[x]>greutate[y])
    {
        parinte[parinte_y]=parinte_x;
        greutate[parinte_x] += greutate[parinte_y];
    }
    else{
        parinte[parinte_x]=parinte_y;
        greutate[parinte_y] += greutate[parinte_x];
    }
}

int main()
{
    int n,m,op,x,y;
    f>>n>>m;
    for(int i=1;i<=n;i++){
        parinte[i]=i;
        greutate[i]=1;
    }
    for(int i=0;i<m;i++)
    {
        f>>op>>x>>y;
        if(op==1)
        {
           unite(x,y);
        }
        else
        {
            if(find(x)==find(y))
                g<<"DA\n";
            else g<<"NU\n";
        }
    }
    return 0;
}