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

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()