0753 - Pre Ordine 1

Cerința

Se consideră un arbore binar în care nodurile memorează numere naturale nenule. Să se afișeze valorile memorate în subarborele stâng al rădăcinii în urma parcurgerii în preordine.

Date de intrare

Fișierul de intrare preordine1in.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 apare apărea 0.

Date de ieșire

Fișierul de ieșire preordine1out.txt va conține pe prima linie valorile memorate în subarborele stâng al rădăcinii în urma parcurgerii în preordine, separate prin exact un spațiu.

Restricții și precizări

• se recomandă folosirea arborilor alocați dinamic.

Exemplul 1:

preordine1in.txt

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

preordine1out.txt

Datele de intrare corespund restrictiilor impuse
51 18 24

Exemplul 2:

preordine1in.txt

preordine1

preordine1out.txt

Datele de intrare nu corespund restrictiilor impuse

Explicație

Exemplul corespunde arborelui desenat mai jos:

Rezolvare

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


def construct_tree(date, index):
    if index >= len(date) or int(date[index]) == 0:
        return None, index + 1

    radacina = Node(int(date[index]))
    radacina.left, index = construct_tree(date, index + 1)
    radacina.right, index = construct_tree(date, index)

    return radacina, index


def preordine(radacina):
    if radacina:
        file_out.write(str(radacina.data) + " ")
        preordine(radacina.left)
        preordine(radacina.right)


def validare(date):
    if len(date) > 1000:
        raise ValueError
    for num in date:
        if not num.isdigit():
            raise ValueError
    file_out.write("Datele de intrare corespund restrictiilor impuse\n")


if __name__ == '__main__':
    file_in = open("preordine1in.txt", "r")
    file_out = open("preordine1out.txt", "w")

    try:
        data = file_in.readline().split()
        validare(data)
        root, _ = construct_tree(data, 0)
        preordine(root.left)

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