Cod sursa(job #2277464)

Utilizator vladuteluVlad Oancea vladutelu Data 6 noiembrie 2018 12:19:58
Problema Paduri de multimi disjuncte Scor 10
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.93 kb
#include <iostream>
#include <fstream>
using namespace std;

ifstream in("disjoint.in");
ofstream out("disjoint.out");

int T[100005];
int S[100005];

int sef(int x)
{
    if(x!=T[x])
        x = sef(T[x]);
    return T[x];
}

void join(int x, int y)
{
    int t1 = sef(x);
    int t2 = sef(y);
    T[t2]=t1;
    S[y]=t1;
}

int main()
{
    int n, m, cod, x, y, i;
    in>>n>>m;
    for(i = 1; i<=n; i++)
        T[i]=i;
    for(i = 1; i<=m; i++)
    {
        in>>cod>>x>>y;
        if(cod==1)
            join(x, y);
        else
        {
            if(S[x]!=0 && S[y]!=0)
            {
                if(S[x]==S[y])
                    out<<"DA"<<endl;
                else out<<"NU"<<endl;
            }
            else
            {
                if(sef(x)==sef(y))
                    out<<"DA"<<endl;
                else out<<"NU"<<endl;
            }
        }
    }
    return 0;
}