Cod sursa(job #2418219)

Utilizator CameliaSSamoilescu Camelia CameliaS Data 4 mai 2019 11:59:55
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.98 kb
#include <iostream>
#include <fstream>
#include <vector>
using namespace std;

ifstream f("disjoint.in");
ofstream g("disjoint.out");

#define NMAX 100005

int paduri[NMAX], rang[NMAX];
int n, m;

int find_father(int x){
    if(paduri[x] == x)
        return x;
    int ans = find_father(paduri[x]);
    paduri[x] = ans;
    return ans;
}

void unio(int x, int y){
    if(rang[x] < rang[y]){
        paduri[x] = y;
        rang[y] += rang[x];
    }
    else
    {
        paduri[y] = x;
        rang[x] += rang[y];
    }
}


int main()
{

    f>>n>>m;
    for(int i = 1; i <= n; i ++)
    {
        rang[i] = 1;
        paduri[i] = i;
    }
    for(int i = 0; i < m; i ++)
    {
        int cod, a, b;
        f>>cod>>a>>b;
        if(cod == 2){
            if(find_father(a) == find_father(b))
                g<<"DA\n";
            else g<<"NU\n";
        }
        else
            unio(find_father(a), find_father(b));
    }
    return 0;
}