Cod sursa(job #2525583)

Utilizator YetoAdrian Tonica Yeto Data 17 ianuarie 2020 16:31:50
Problema Paduri de multimi disjuncte Scor 70
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.79 kb
#include <fstream>
using namespace std;
int n, m, i, x, y, z, tx, ty;
int tata[100001];
ifstream fin ("disjoint.in");
ofstream fout ("disjoint.out");

int parinte (int x)
{
    while (tata[x]>0)
        x=tata[x];
    return x;
}

int main () {
    fin>>n>>m;
    for (i=1;i<=n;i++)
        tata[i]=-1;

    for (i=1;i<=m;i++) {
        fin>>z>>x>>y;
        tx=parinte(x);
        ty=parinte(y);
        if (z==2) {
            if (tx!=ty)
                fout<<"NU\n";
            else
                fout<<"DA\n";
        } else if (z==1){
            if (tx<ty) {
                tata[tx]+=tata[ty];
                tata[ty]=tx;
            } else {
                tata[ty]+=tata[tx];
                tata[tx]=ty;
            }
        }
    }
    return 0;
}