Editing 0758 - Bi Min Prim

== Cerinţa ==
Se consideră un arbore binar în care nodurile memorează numere naturale nenule. Să se determine cele mai mici valori număr prim din subarborii stâng și drept ai rădăcinii.
== Date de intrare ==
Fișierul de intrare '''biminprimin.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 '''biminprimout.txt''' va conține pe prima linie două valori '''X''' '''Y''', reprezentând cea mai mică valoare număr prim din subarborele stâng, respectiv cea mai mică valoare număr prim din subarborele drept. Dacă unul dintre subarbori nu conține numere prime, se va afișa '''-1'''.
== Restricţii şi precizări ==
* se recomandă folosirea arborilor alocați dinamic.
* se garantează că rădăcina are doi descendenți direcți
== Exemplul 1 ==
; '''biminprimin.txt'''
 67 53 17 0 0 24 0 0 48 0 12 0 0
; '''biminprimout.txt'''
 Datele de intrare corespund restrictiilor impuse
 17 -1
== Exemplul 2 ==
; '''biminprimin.txt'''
 1 0 0
; '''biminprimout.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 is_prime(n):
    if n < 2:
        return False
    for i in range(2, int(n**0.5) + 1):
        if n % i == 0:
            return False
    return True


def find_min_prime(node):
    if node is None:
        return float('inf')
    left = find_min_prime(node.left)
    right = find_min_prime(node.right)
    current = node.value if is_prime(node.value) else float('inf')
    return min(left, right, current)


def main():
    with open('biminprimin.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")
    left_min_prime = find_min_prime(root.left)
    right_min_prime = find_min_prime(root.right)
    left_min_prime = -1 if left_min_prime == float('inf') else left_min_prime
    right_min_prime = -1 if right_min_prime == float('inf') else right_min_prime
    with open('biminprimout.txt', 'w') as fout:
        fout.write(f"{left_min_prime} {right_min_prime}")


if __name__ == "__main__":
    main()
</syntaxhighlight>