Cod sursa(job #1181125)

Utilizator DuxarFII-Stefan-Negrus Duxar Data 1 mai 2014 21:17:26
Problema Paduri de multimi disjuncte Scor 70
Compilator cpp Status done
Runda Arhiva educationala Marime 2.02 kb
#include <fstream>
#include <iostream>
#include <string>
#include <complex>
#include <cmath>
#include <set>
#include <vector>
#include <map>
#include <queue>
#include <cstdio>
#include <stack>
#include <algorithm>
#include <list>
#include <bitset>
#include <ctime>
#include <cassert>
#include <iomanip>

using namespace std;


const string file = "disjoint";
const string inputF = file + ".in";
const string outputF = file + ".out";

const double epsilon = 1e-7;
#define LL long long
#define ULL unsigned long long 
#define MOD1 666013
#define MOD2 666019
#define MOD3 1999999973
#define base 73

typedef vector<int> vi;
typedef vector<long long> vl;
typedef vector<bool> vb;
typedef vector<vi> vvi;
typedef vector<vl> vvl;
typedef vector<double> vd;
typedef vector<string> vs;
typedef pair<int, int> ii;
typedef pair<long long, long long> ll;
typedef vector<ii> vii;
typedef vector<ll> vll;

#define all(V) V.begin(), V.end()
#define allr(V) V.rbegin(), V.rend()
#define for_c_it(container, it) for (auto it : container)
#define present(container, element) (container.find(element) != container.end()) 
#define sz(a) int((a).size()) 
#define pb push_back 
#define mp make_pair
#define zeroes(x) ((x ^ (x - 1)) & x)

int N, Q;
int father[100001];

void update(int x, int y);
string query(int x, int y);

int main() {
#ifndef INFOARENA
	freopen("input.txt", "r", stdin);
#else 
	freopen(inputF.c_str(), "r", stdin);
	freopen(outputF.c_str(), "w", stdout);
#endif
	int i, x, y, cod;
	
	scanf("%d %d", &N, &Q);
	//father.resize(N + 1);
	
	for (i = 0; i < Q; ++i) {
		scanf("%d %d %d", &cod, &x, &y);
		if (cod == 1) {
			update(x, y);
		}
		else {
			printf("%s\n", query(x,y).c_str());
		}
	}		

	return 0;
}

void update(int x, int y) {
	int lgx = 1, lgy = 1;
	for (; father[x]; x = father[x], ++lgx);
	for (; father[y]; y = father[y], ++lgy);
	if (lgx <= lgy) {
		father[x] = y; 
	}
	else {
		father[y] = x;
	}
	
}

string query(int x, int y) {
	for (; father[x]; x = father[x]);
	for (; father[y]; y = father[y]);
	if (x == y) {
		return "DA";
	}
	return "NU";
}