3235 - Entries

Enunț

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 entriesIN.txt 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 entriesOUT.txt 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.În cazul în care restricțiile nu sunt îndeplinite, se va afișa mesajul "Nu corespunde restricțiilor impuse".

Restricții și precizări

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

Exemplul 1:

entriesIN.txt

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

entriesOUT.txt

0
0
1
0
1
0

Exemplul 2:

entriesIN.txt

5001
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

entriesOUT.txt

Datele nu corespund restrictiilor impuse

Rezolvare

def verifica_restrictiile(n, p):
    if not (1 <= n <= 5000):
        raise ValueError("")
    if not (1 <= p <= 10000000):
        raise ValueError("")

def main():
    fin = open("entriesIN.txt", "r")
    fout = open("entriesOUT.txt", "w")

    M = {}
    t = [0] * 10005
    n = 0

    def find(x):
        z, y = x, x
        while t[x] != 0:
            x = t[x]

        while t[y] != 0:
            z = t[y]
            t[y] = x
            y = z

        return x

    def union(x, y):
        t[y] = x

    T = int(fin.readline().strip())
    try:
        verifica_restrictiile(T, T)  # Check restrictions for n (number of test cases)
    except ValueError as e:
        fout.write(f"Datele nu corespund restrictiilor impuse {str(e)}")
        fin.close()
        fout.close()
        return

    for _ in range(T):
        x, y, op = map(int, fin.readline().strip().split())
        k = 0
        if x not in M:
            k = 1
            n += 1
            M[x] = n
        if y not in M:
            k = 1
            n += 1
            M[y] = n

        x = M[x]
        y = M[y]
        x = find(x)
        y = find(y)

        if op == 1:
            if x != y:
                union(x, y)
        else:
            if k == 1:
                fout.write("0\n")
            elif x == y:
                fout.write("1\n")
            else:
                fout.write("0\n")

    fin.close()
    fout.close()

if __name__ == "__main__":
    main()