Cod sursa(job #2942689)

Utilizator simonatudoroiuTudoroiu Simona simonatudoroiu Data 19 noiembrie 2022 22:53:35
Problema Paduri de multimi disjuncte Scor 10
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.23 kb
#include <bits/stdc++.h>
using namespace std;
ifstream fin("disjoint.in");
ofstream fout("disjoint.out");
int main()
{
    int n,m,a,b,c;
    vector<vector<int>> list;
    vector<int> tata;
    fin>>n>>m;
    for(int i=0;i<=n;i++)
    {
        tata.push_back(i);
        list.push_back({});
    }
    for(int x=1;x<=m;x++)
    {
        fin>>a>>b>>c;
        if(a == 1)
        {
            if(list[tata[c]].size() < list[tata[b]].size()) {
                list[tata[b]].push_back(tata[c]);
                tata[c] = tata[b];
                for (int i: list[tata[c]]) {
                    list[tata[b]].push_back(i);
                    tata[i] = tata[b];
                }
                list[tata[c]].clear();
            }
            else
            {
                list[tata[c]].push_back(tata[b]);
                tata[b] = tata[c];
                for (int i: list[tata[b]]) {
                    list[tata[c]].push_back(i);
                    tata[i] = tata[c];
                }
                list[tata[b]].clear();
            }
        }
        else
        {
            if(tata[b] == tata[c])
                fout<<"DA\n";
            else
                fout<<"NU\n";
        }

    }
    fin.close();
    fout.close();
    return 0;
}