0759 - Bi Latime

Cerinţa

Se consideră un arbore binar în care nodurile memorează numere naturale nenule. Să se afișeze valorile din arbore în urma parcurgerii în lățime, pornind din rădăcină.

Date de intrare

Fișierul de intrare bilatimein.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 apare valoarea 0. Dacă un nod nu are descendent drept, în listă va apare valoarea 0.

Date de ieșire

Fișierul de ieșire bilatimeout.txt va conține pe prima linie valorile din arbore în urma parcurgerii în lățime, pornind din rădăcină, separate prin câte un spațiu.

Restricţii şi precizări

• se recomandă folosirea arborilor alocați dinamic.
• la parcurgerea în lățime, după vizitarea unui nod, se va vizita fiul stâng, apoi fiul drept.

Exemplul 1

bilatimein.txt

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

bilatimeout.txt

Datele de intrare corespund restrictiilor impuse
67 51 48 18 24 11

Exemplul 2

bilatimein.txt

1 0 0

bilatimeout.txt

Datele de intrare nu corespund restrictiilor impuse

Rezolvare

<syntaxhighlight lang="python" line> class Node:

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

def build_tree(values):

   value = next(values)
   if value == 0:
       return None
   node = Node(value)
   node.left = build_tree(values)
   node.right = build_tree(values)
   return node

def breadth_first_search(root):

   queue = [root]
   values = []
   while queue:
       node = queue.pop(0)
       if node is not None:
           values.append(node.value)
           queue.append(node.left)
           queue.append(node.right)
   return values

def main():

   with open('bilatimein.txt', 'r') as fin:
       values = list(map(int, fin.readline().split()))
   root = build_tree(iter(values))
   if root is None or root.left is None or root.right is None:
       print("Datele de intrare nu corespund restrictiilor impuse")
       return
   print("Datele de intrare corespund restrictiilor impuse")
   bfs_values = breadth_first_search(root)
   with open('bilatimeout.txt', 'w') as fout:
       fout.write(' '.join(map(str, bfs_values)))

if __name__ == "__main__":

   main()

</syntaxhighlight>