Cod sursa(job #2003527)

Utilizator MaligMamaliga cu smantana Malig Data 23 iulie 2017 09:05:34
Problema Paduri de multimi disjuncte Scor 10
Compilator cpp Status done
Runda Arhiva educationala Marime 1.39 kb
#include <iostream>
#include <fstream>
#include <cstdlib>
#include <time.h>
#include <algorithm>

using namespace std;
ifstream in("disjoint.in");
ofstream out("disjoint.out");

#define ll long long
#define ui unsigned int
#define pb push_back
const int inf = 1e9 + 5;
const int NMax = 1e6 + 5;

int N,M;
int nrArb[NMax],dad[NMax];

int findRoot(int);
void unite(int,int);

int main() {
    in>>N>>M;

    for (int i=1;i <= N;++i) {
        dad[i] = i;
        nrArb[i] = 1;
    }

    while (M--) {
        int tip,x,y;
        in>>tip>>x>>y;

        switch(tip) {
        case 1: {
            unite(x,y);
            break;
        }
        case 2: {
            int root1 = findRoot(x), root2 = findRoot(y);
            if (root1 == root2) {
                out<<"DA\n";
            }
            else {
                out<<"NU\n";
            }

            break;
        }
        }
    }

    in.close();
    out.close();
    return 0;
}

int findRoot(int node) {
    if (dad[node] == node) {
        return node;
    }

    int root = findRoot(dad[node]);
    dad[node] = root;
    return root;
}

void unite(int x,int y) {
    int root1 = findRoot(x), root2 = findRoot(y);
    if (nrArb[x] > nrArb[y]) {
        dad[y] = x;
        nrArb[x] += nrArb[y];
    }
    else {
        dad[x] = y;
        nrArb[y] += nrArb[x];
    }
}