Cod sursa(job #1394406)

Utilizator prisonbreakMichael Scofield prisonbreak Data 20 martie 2015 12:03:35
Problema Paduri de multimi disjuncte Scor 40
Compilator cpp Status done
Runda Arhiva educationala Marime 0.94 kb
#include <iostream>
#include <fstream>
#include <cstdlib>
#include <ctime>
 
using namespace std;
 
//ifstream f("disjoint.in");
ofstream g("disjoint.out");
 
int N,M,x,y,i,cod,t[100005],rang[100005];
 
 
inline int find_comp(int x)
{
    if(x!=t[x])
    {
        int tata=find_comp(t[x]);
        t[x]=tata;
    }
    return t[x];
}
 
inline void unite(int x, int y)
{
    int tx=find_comp(x);
    int ty=find_comp(y);
    if(rang[tx]<=rang[ty])
        { t[tx]=ty; rang[ty]++; }
        else
            { t[ty]=tx; rang[tx]++; }
}
 
inline void test(int x, int y)
{
    if(find_comp(x)==find_comp(y))
        g<<"DA"<<endl;
    else
        g<<"NU"<<endl;
}
 
 
int main()
{
    ifstream fin("disjoint.in");
 
    fin >> N >> M;
    for(i=1;i<=N;i++)
        t[i]=i;
    for(i=1;i<=M;i++)
    {
    	fin >> cod >> x >> y;
        //scanf("%d%d%d", &cod, &x, &y);
        if(cod==1)
            unite(x,y);
        if (cod == 2)
            test(x,y);
    }
}