Cod sursa(job #2823517)

Utilizator gianikiritaGiani Kirita gianikirita Data 28 decembrie 2021 18:47:19
Problema Paduri de multimi disjuncte Scor 40
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.06 kb
#include <fstream>

using namespace std;

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

int v[100005];

void init(int n)
{
    for(int i=1;i<=n;i++)
    {
        v[i]=i;
    }
}

int find(int a)
{
    if(v[a]==a)
    {
        return a;
    }
    return find(v[a]);
}//gasim varful grafului care il contine pe a

void unite(int a,int b)
{
    v[find(b)]=find(a);
}//legam varful grafului care il contine pe b de varful grafului care il contine pe a
int main()
{
    int n,m,c,a,b;
    fin>>n>>m;//nr de multimi ,nr de operatii
    init(n);
    for(int i=1;i<=m;i++)//fiecare operatie
    {
        fin>>c>>a>>b;
        if(c==1)
        {
            unite(a,b);
        }
        else
        {
            if(find(a)==find(b))//verificam daca varful grafului care il contine pe a este acelasi cu varful grafului care il contine pe b
            {
                fout<<"DA"<<endl;
            }
            else
            {
                fout<<"NU"<<endl;
            }
        }

    }
    return 0;
}