Cod sursa(job #2823621)

Utilizator gianikiritaGiani Kirita gianikirita Data 29 decembrie 2021 01:26:55
Problema Paduri de multimi disjuncte Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.36 kb
#include <fstream>
#include <vector>

using namespace std;

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

struct DSU
{
    int N;
    vector<int> parent;

    void init(int n)
    {
   //     N=n;
   //     parent.resize(N+1);//formam un vector de n+1 elemente
        for(int i=1; i<=N; i++)
        {
            parent[i]=i;
        }
    }

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

    void unite(int a,int b)
    {
        a=find(a);
        b=find(b);
        parent[b]=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
    struct DSU dsu;
    dsu.init(n);
    for(int i=1; i<=m; i++) //fiecare operatie
    {
        fin>>c>>a>>b;
        if(c==1)
        {
            dsu.unite(a,b);
        }
        else
        {
            if(dsu.find(a)==dsu.find(b))//verificam daca varful grafului care il contine pe a este acelasi cu varful grafului care il contine pe b
            {
                fout<<"DA"<<"\n";
            }
            else
            {
                fout<<"NU"<<"\n";
            }
        }

    }
    return 0;
}