2160 - Prize

Am câștigat un concurs și primim la școală foarte foarte multe echipamente pentru un nou laborator. Pentru amenajarea laboratorului am primit și o sală de clasă. Din păcate în sala de clasă există doar o singură priză, în care ar putea fi conectat doar un singur echipament. Cum nu putem reface imediat instalația electrică, am hotărât să utilizăm prelungitoare.

Un prelungitor poate avea una sau mai multe prize în care pot fi conectate echipamente și eventual alte prelungitoare. Evident, pentru ca prelungitorul să poată fi utilizat el trebuie să fie alimentat la curentul electric.

Cerința

Cunoscând configurația prelungitoarelor să se determine numărul maxim de echipamente ce pot fi alimentate la curentul electric.

Date de intrare

Fișierul de intrare prize.in conține pe prima linie numărul natural N, care reprezintă numărul de prelungitoare. Pe următoarele N linii se află N numere naturale nenule, câte un număr pe o linie, reprezentând numărul de prize din fiecare dintre cele N prelungitoare.

Date de ieșire

Fișierul de ieșire prize.out va conține o singură linie pe care va fi scris numărul maxim de echipamente ce pot fi alimentate la curent, utilizând cele N prelungitoare descrise în fișierul de intrare.

Restricții și precizări

1 ≤ N ≤ 1000 1 ≤ numărul de prize dintr-un prelungitor ≤ 5 000 000

Exemplu:

prize.in 3 3 2 5

prize.out 8

Explicație

O modalitate de conectare a prelungitoarelor pentru a alimenta la curent echipamente ar fi:

punem în priză prelungitorul 3; prelungitorul 1 îl punem în una dintre prizele prelungitorului 3 (mai rămân în prelungitorul 3 patru prize libere pentru echipamente) prelungitorul 2 îl punem în una dintre prizele prelungitorului 1 (mai rămân în prelungitorul 1 două prize libere pentru echipamente). În total vor exista 4+2+2=8 prize disponibile pentru echipamente.

Rezolvare

<syntaxhighlight lang="python" line> def validate(n: int, prizes: list) -> bool:

   """
   Verifică dacă datele de intrare sunt valide.
   """
   if n != len(prizes):
       return False
   for i in range(n):
       if prizes[i] <= 0:
           return False
   return True

def read_input_file(file_path: str) -> tuple:

   """
   Citește datele de intrare din fișierul specificat.
   """
   with open(file_path, 'r') as f:
       n = int(f.readline())
       prizes = []
       for _ in range(n):
           prizes.append(int(f.readline()))
       return n, prizes

def find_max_devices(n: int, prizes: list) -> int:

   """
   Găsește numărul maxim de echipamente ce pot fi alimentate la curentul electric.
   """
   max_devices = 0
   for i in range(n):
       devices = prizes[i]
       for j in range(i):
           if prizes[j] > 0:
               devices += prizes[j] - 1
       if devices > max_devices:
           max_devices = devices
   return max_devices

def main():

   input_file = 'prize.in'
   output_file = 'prize.out'

   n, prizes = read_input_file(input_file)
   if not validate(n, prizes):
       print("Date de intrare invalide")
       return

   max_devices = find_max_devices(n, prizes)

   with open(output_file, 'w') as f:
       f.write(str(max_devices))

if __name__ == '__main__':

   main()

</syntaxhighlight>

Explicatie

Funcția validate verifică dacă datele de intrare sunt valide. Verifică dacă numărul de prelungitoare este egal cu lungimea listei prizes și dacă fiecare număr din listă este pozitiv.

Funcția read_input_file citește datele de intrare din fișierul specificat. Citeste mai întâi numărul de prelungitoare și apoi citește numărul de prize pentru fiecare prelungitor.

Funcția find_max_devices găsește numărul maxim de echipamente ce pot fi alimentate la curentul electric. Folosim două bucle for pentru a itera prin toate prelungitoarele și pentru fiecare prelungitor calculăm numărul maxim de echipamente care pot fi conectate în prelungitor. Adăugăm la acest număr de prize numărul de prize din prelungitoarele anterioare (din listă) minus 1, deoarece numai una dintre prizele acestora poate fi utilizată pentru a conecta prelungitorul curent.

În funcția main, citim datele de intrare din fișierul prize.in, verificăm dacă acestea sunt valide și apoi apelăm funcția find_max_devices pentru a obține numărul maxim de echipamente ce pot fi alimentate la curent. Rezultatul este scris în fișierul prize.out.