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 validare(n, nums):

   if not (1 <= n <= 1000):
       return False
   if not all(0 <= nums <= 5_000_000 for nums in nums):
       return False
   return True

def prize(n, nums):

   s = sum(nums)
   fout.write(str(s - (n - 1)))
   fin.close()
   fout.close()

if __name__ == '__main__':

   fin = open("prize.in")
   fout = open("prize.out", "w")

   lines = fin.readlines()
   n = int(lines[0].strip())
   nums = list(map(int, lines[1:]))
   if validare(n, nums):
       print("Date de intrare valide")
       prize(n, nums)
   else:
       print("Date de intrare invalide")

   fin.close()
   fout.close()

</syntaxhighlight>