Cod sursa(job #2390705)

Utilizator irinaS13Sandu Irina irinaS13 Data 28 martie 2019 11:42:40
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.86 kb
#include <fstream>
#include <vector>
using namespace std;
ifstream fin("disjoint.in");
ofstream fout("disjoint.out");
int comp[100001],n;
vector <vector<int>> nod_comp;
void reuniune(int x, int y)
{
    x=comp[x];y=comp[y];
    int nrx=nod_comp[x].size(),nry=nod_comp[y].size();
    if(nrx<nry) swap(x,y);
    for(auto i:nod_comp[y])
        {comp[i]=x;
         nod_comp[x].push_back(i);}
    nod_comp[y].clear();

}
int main()
{
    int m;
    fin>>n>>m;
    nod_comp= vector <vector<int> > (n+1);
    for(int i=1;i<=n;i++)
    {
        comp[i]=i;
        nod_comp[i].push_back(i);
    }
    for(int i=1;i<=m;i++)
    {
        int cod,x,y;
        fin>>cod>>x>>y;
        if(cod==1) reuniune(x,y);
        if(cod==2)
        {
            if(comp[x]==comp[y]) fout<<"DA\n";
            else fout<<"NU\n";
        }
    }

    return 0;
}