fbpx

Problema #3235 – entries – Rezolvari PBInfo

de Mihai-Alexandru

Se consideră un graf care inițial este format din P noduri izolate, etichetate de la 1 la P. Se mai consideră N intrări, unde intrare poate însemna:

  • comandă – o comandă are forma I + J, cu semnificația că în graf se adaugă muchia care unește nodurile I și J (dacă I și J erau deja unite în acel moment, nu se întreprinde nici o acțiune);
  • întrebare – o întrebare este de forma I ? J, adică se întreabă dacă în acel moment I și J sunt în aceeași componentă conexă.

Se pleacă deci de la un graf inițial format din noduri izolate, care pe parcurs se “unifică”. Tot pe parcurs sunteți întrebat dacă anumite perechi de noduri sunt sau nu în aceeași componentă conexă.

Cerința

Scrieți un program care să răspundă la întrebările din fișierul de intrare.

Date de intrare

Din fișierul entries.in veți citi de pe prima linie numărul N de intrări. Pe următoarele N linii se găsesc intrările, câte una pe linie. O intrare este codificată prin trei numere separate prin câte un blanc. Primele două numere reprezintă nodurile I și J (numere întregi, cuprinse între 1 și P), iar al treilea este 1 dacă intrarea este o comandă, respectiv 2 dacă intrarea este o întrebare.

Date de ieșire

Fișierul de ieșire entries.out va conține răspunsurile la întrebări, câte un răspuns pe o linie, în ordinea din fișierul de intrare. Răspunsul va fi numărul 1 dacă nodurile despre care ați fost întrebat sunt în acel moment în aceeași componentă conexă, respectiv numărul 0 în caz contrar.

Restricții și precizări

  • 1 ≤ N ≤ 5 000
  • 1 ≤ P ≤ 10.000.000

Exemplu

entries.in

9
1 2 2 
1 2 1
3 7 2
2 3 1
1 3 2
2 4 2
1 4 1
3 4 2
1 7 2

entries.out

0
0
1
0
1
0
#include <bits/stdc++.h>

using namespace std;

ifstream cin("entries.in");
ofstream cout("entries.out");

int T[10001] , n , x , y , cer;
unordered_map <int , int> M;

void leaga(int a , int b)
{
    if(T[a] > T[b]) T[a] = T[b];
    else T[b] = T[a];
}

int radacina(int a)
{
    if(a == T[a]) return a;
    else return T[a] = radacina(T[a]);
}

int main()
{
    cin >> n;
    for(int i = 1 ; i <= 10000 ; i++)
        T[i] = i;
    for(int i = 1 ; i <= n ; i++)
    {
        cin >> x >> y >> cer;

        if(M[x] == 0) M[x] = M.size();
        if(M[y] == 0) M[y] = M.size();

        x = M[x] , y = M[y];

        if(cer == 1)
        {
            int r1 = radacina(x) , r2 = radacina(y);
            if(r1 != r2) leaga(r1 , r2);
        }

        else cout << (radacina(x) == radacina(y)) << '\n';
    }
}
Comentarii

S-ar putea sa iti placa