Cod sursa(job #2170112)

Utilizator dragosmdvMoldovan Dragos dragosmdv Data 14 martie 2018 22:25:29
Problema Paduri de multimi disjuncte Scor 70
Compilator cpp Status done
Runda Arhiva educationala Marime 0.99 kb
#include <iostream>
#include <fstream>

using namespace std;

ifstream fin("disjoint.in");
ofstream fout("disjoint.out");

int v[100010], n, m, k, a, b;

int find_(int x)
{
    if(v[x]<0)
        return x;
    else
        return x=find_(v[x]);
}
void union_(int a, int b)
{
    int parenta=find_(a);
    int parentb=find_(b);
    if(v[parenta]<v[parentb])
        v[parenta]=parentb;
    else if(v[parenta]>v[parentb])
        v[parentb]=parenta;
    else if(v[parenta]==v[parentb])
{
        v[parentb]=parenta;
        v[parenta]--;
    }
}

int main()
{
    fin>>n>>m;
    for(int i=1; i<=n; i++)
        v[i]=-1;
    for(int i=1; i<=m; i++)
    {
        fin>>k;
        if(k==1)
        {
            fin>>a>>b;
            union_(a,b);
        }
        else if(k==2)
        {
            fin>>a>>b;
            if(find_(a)==find_(b))
                fout<<"DA\n";
            else
                fout<<"NU\n";
        }
    }

    return 0;
}