0792 - Bi NeFrunze 1

Cerința

Se consideră un arbore binar în care nodurile memorează numere naturale nenule. Să se afișeze valorile memorate în nodurile neterminale ale arborelui, în ordine descrescătoare.

Date de intrare

Fișierul de intrare binefrunze1in.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 binefrunze1out.txt va conține pe prima linie valorile memorate în nodurile neterminale ale arborelui, în ordine descrescătoare, separate prin câte un spațiu.

Restricții și precizări

• se recomandă folosirea arborilor alocați dinamic.
• arborele nu va avea mai mult de 5000 de noduri neterminale

Exemplul 1:

binefrunze1in.txt

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

binefrunze1out.txt

Datele de intrare corespund restrictiilor impuse
67 51 48

Exemplul 2:

binefrunze1in.txt

binefrunze1

binefrunze1out.txt

Datele de intrare nu corespund restrictiilor impuse

Explicație

Exemplul corespunde arborelui desenat mai jos:

Rezolvare

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


def construct_tree(data_list):
    data = data_list.pop(0)
    if data == 0:
        return None
    node = Node(data)
    node.left = construct_tree(data_list)
    node.right = construct_tree(data_list)
    return node


def get_non_leaf_nodes(node, non_frunze):
    if node is None or (node.left is None and node.right is None):
        return
    non_frunze.append(node.data)
    get_non_leaf_nodes(node.left, non_frunze)
    get_non_leaf_nodes(node.right, non_frunze)


def validare(nr):
    if len(nr) > 5000:
        raise ValueError
    for cifra in nr:
        if cifra < 0:
            raise ValueError
    file_out.write("Datele de intrare corespund restrictiilor impuse\n")


if __name__ == '__main__':
    file_in = open("binefrunze1in.txt", "r")
    file_out = open("binefrunze1out.txt", "w")
    try:
        numar = list(map(int, file_in.readline().split()))
        validare(numar)
        root = construct_tree(numar)
        non_leaf_nodes = []
        get_non_leaf_nodes(root, non_leaf_nodes)
        non_leaf_nodes.sort(reverse=True)
        file_out.write(' '.join(map(str, non_leaf_nodes)))
    except ValueError:
        file_out.write("Datele de intrare nu corespund restrictiilor impuse")
    except IndexError:
        file_out.write("Datele de intrare nu corespund restrictiilor impuse")