Cod sursa(job #642984)

Utilizator johnny2008Diaconu Ion johnny2008 Data 2 decembrie 2011 18:04:20
Problema Paduri de multimi disjuncte Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 1.01 kb
#include<fstream>
#include<iostream>
#include<string>
#include<vector>
using namespace std;
#define pb push_back
int n,m;
int apart[100001];
vector<vector <int> > vec;
int main(){
	ifstream f("disjoint.in");
	ofstream g("disjoint.out");
	f>>n>>m;
	int i,j;
	int a,x,y;
	
	for(i=1;i<=n;i++){
		vec[i].pb(1);
		vec[i].pb(i);
		
		apart[i]=i;
	}
	
	for(i=1;i<=m;i++){
		f>>a>>x>>y;
		
		if(a==1){
			/*
			for(j=1;j<=n;j++)
				cout<<apart[j]<<" ";
			cout<<"\n";
			for(j=1;j<=n;j++){
				for(int q=1;q<=vec[j][0];q++)
					cout<<vec[j][q]<<" ";
				cout<<"\n";
			}
			*/
			int vecx=apart[x];
			int vecy=apart[y];
			for(j=1;j<=vec[vecy][0];j++){
				apart[vec[vecy][j]]=vecx;
			}
			for(j=vec[vecx][0]+1;j<=vec[vecx][0]+vec[vecy][0];j++){
				
				vec[vecx].pb(vec[vecy][j-vec[vecx][0]]);
				
			}
			vec[vecx][0]=vec[vecx][0]+vec[vecy][0];
			
			
		}
		else{
			if(apart[x]==apart[y]){
				g<<"DA\n";
			}
			else{
				g<<"NU\n";
			}
		}
		
	}
	
	return 0;
}