Cod sursa(job #804914)

Utilizator SebiSebiPirtoaca George Sebastian SebiSebi Data 30 octombrie 2012 17:46:44
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 0.66 kb
#include<iostream>
#include<fstream>
using namespace std;

#define NMAX 100001

int p[NMAX];

int multime(int x)
{
	int r,y;
	r=x;
	while(p[r]!=r)
		r=p[r];
	while(p[x]!=x) {
		y=p[x];
		p[x]=r;
		x=y;
	}
	return r;
}

void reuneste(int x, int y)
{
	if(x>y)
		swap(x,y);
	p[y]=x;
}

int main ()
{
	int n,m,i,x,y,opt;
	ifstream f("disjoint.in");
	ofstream g("disjoint.out");
	f>>n>>m;
	for(i=1;i<=n;i++)
		p[i]=i;
	for(i=1;i<=m;i++) {
		f>>opt>>x>>y;
		if(opt==1)
			reuneste(multime(x),multime(y));
		else {
			if(multime(x)==multime(y))
				g<<"DA";
			else g<<"NU";
			g<<'\n';
		}
	}
	f.close();
	g.close();
	return 0;
}