Cod sursa(job #1138887)

Utilizator UAIC_Balan_Negrus_HreapcaUAIC Balan Negrus Hreapca UAIC_Balan_Negrus_Hreapca Data 10 martie 2014 18:12:12
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.86 kb
#include <vector>
#include <list>
#include <map>
#include <set>
#include <deque>
#include <queue>
#include <stack>
#include <bitset>
#include <algorithm>
#include <functional>
#include <numeric>
#include <utility>
#include <sstream>
#include <iostream>
#include <iomanip>
#include <cstdio>
#include <cmath>
#include <cstdlib>
#include <cctype>
#include <string>
#include <cstring>
#include <cstdio>
#include <cmath>
#include <cstdlib>
#include <ctime>
#include <fstream>
#include <iterator>
#include <random>
#include <assert.h>
using namespace std;

const string file = "disjoint";

const string infile = file + ".in";
const string outfile = file + ".out";

const int INF = 0x3f3f3f3f; 

//#define ONLINE_JUDGE

vector<int> tata;
vector<int> grad;

int Find(int x)
{
    int t = x;
    while(tata[t] != 0)
        t = tata[t];

    int prec;
    for(int i = x; i != t; i = prec)
    {
        prec = tata[i];
        tata[i] = t;
    }
    return t;
}
void Union(int x, int y)
{
    if(grad[x] < grad[y])
        swap(x, y);
    tata[y] = x;
    grad[x] += grad[y];
}

int main()
{
#ifdef ONLINE_JUDGE
	ostream &fout = cout;
	istream &fin = cin;
#else
	fstream fin(infile.c_str(), ios::in);
	fstream fout(outfile.c_str(), ios::out);
#endif	

    int N, M;
    fin >> N >> M;
    tata.resize(N + 1, 0);
    grad.resize(N + 1, 1);
    for(int i = 0; i < M; i++)
    {
        int op, x, y;
        fin >> op >> x >> y;
        if(op == 1)
        {
            int a = Find(x);
            int b = Find(y);
            if(a != b)
                Union(a, b);
        }
        else if(op == 2)
        {
            int a = Find(x);
            int b = Find(y);
            fout << (a == b ? "DA" : "NU") << "\n";
        }
    }

#ifdef ONLINE_JUDGE
#else
    fout.close();
	fin.close();
#endif
}