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>