Cod sursa(job #2428126)

Utilizator georgeoctavianGeorge Octavian Grumazescu georgeoctavian Data 3 iunie 2019 21:11:46
Problema Paduri de multimi disjuncte Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.91 kb
#include <bits/stdc++.h>
using namespace std;
ifstream fin("disjoint.in");
ofstream fout("disjoint.out");
int n,m,x,y,l[100010],mult[100010],tip;
vector <int> v[100010];
void unite(int x, int y)
{
    for(int i=l[y]-1;i>=0;i--)
    {
        v[x].push_back(v[y][i]);
        mult[v[y][i]]=x;
        v[y].pop_back();
    }
    l[x]+=l[y];
    l[y]=0;
}
int main()
{
    ios_base::sync_with_stdio(false);
    fin>>n>>m;
    for(int i=1;i<=n;i++)
    {
        l[i]=1;
        mult[i]=i;
        v[i].push_back(i);
    }
    for(int i=1;i<=m;i++)
    {
        fin>>tip>>x>>y;
        if(tip==1)
        {
            if(l[mult[x]]>l[mult[y]])
                unite(mult[x],mult[y]);
            else unite(mult[y],mult[x]);
        }
        else
        {
            if(mult[x]==mult[y])
                fout<<"DA\n";
            else fout<<"NU\n";
        }
    }
    return 0;
}