3503 - Bi Arbore 1

Cerința

Se consideră un arbore binar în care nodurile memorează numere naturale nenule. Să se determine valorile memorate în descendenții direcți ai rădăcinii arborelui.

Date de intrare

Fișierul de intrare biarbore1in.txt conține pe prima linie lista valorilor memorate în nodurile arborelui, obținute în urma parcurgerii în preordine (rădăcină, stâng, drept). Dacă un nod nu are descendent stâng, în listă va apărea valoarea 0. Dacă un nod nu are descendent drept, în listă va apărea valoarea 0.

Date de ieșire

Fișierul de ieșire biarbore1out.txt va conține pe prima linie două valori X Y, separate prin exact un spațiu; X reprezintă valoarea memorată în fiul stâng al rădăcinii, iar Y reprezintă valoarea memorată în fiul drept al rădăcinii. Dacă rădăcina nu are fiu stâng sau fiu drept, se va afișa corespunzător valoarea 0.

Restricții și precizări

• se vor citi cel mult 1.111.111 numere
• numerele din fișierului de intrare vor fi mai mici decât 1.111.111.111

Exemplul 1:

biarbore1in.txt

67 51 18 0 0 24 0 0 48 0 11 0 0

biarbore1out.txt

Datele de intrare corespund restrictiilor impuse
51 48

Exemplul 2:

biarbore1in.txt

biarbore1

biarbore1out.txt

Datele de intrare nu corespund restrictiilor impuse

Explicație

Exemplul corespunde arborelui desenat mai jos:

Rezolvare

<syntaxhighlight lang="python" line="1" start="1">

class Node:

   def __init__(self, data):
       self.left = None
       self.right = None
       self.data = data

def validare(arbori): # functia de validare a datelor de intrare

   if len(arbori) > 1111111:   # se vor citi cel mult 1.111.111 numere
       raise ValueError

   for nod in arbori:
       if not nod.isdigit() or int(nod) >= 1111111111:    # numerele din fișierului de intrare vor fi mai mici decât 1.111.111.111
           raise ValueError

   file_out.write("Datele de intrare corespund restrictiilor impuse\n")

def biarbore(arbori): # functia de rezolvare

   radacina = Node(arbori[0])
   radacina.left = Node(arbori[1])
   radacina.right = Node(arbori[8])
   file_out.write(f"{radacina.left.data} {radacina.right.data}")

if __name__ == '__main__':

   file_in = open("biarbore1in.txt", "r")         # declararea fisierelor
   file_out = open("biarbore1out.txt", "w")       # fisierul out trebuie declarat cu optiunea "w" (write)

   try:
       arbore = file_in.readline().split()      # citirea arborelui se face ca sir de caractere

       validare(arbore)                 # apelul functiei de validare
       biarbore(arbore)               # apelul functiei de rezolvare

   except ValueError:
       file_out.write("Datele de intrare nu corespund restrictiilor impuse")
   except IndexError:
       file_out.write("Datele de intrare nu corespund restrictiilor impuse")

</syntaxhighlight>