Editing 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 <span style=“color: red”> prize.in</span> conține pe prima linie numărul natural <span style=“color: red”> N</span>, care reprezintă numărul de prelungitoare. Pe următoarele <span style=“color: red”> N</span> linii se află <span style=“color: red”> N</span> numere naturale nenule, câte un număr pe o linie, reprezentând numărul de prize din fiecare dintre cele <span style=“color: red”> N</span> prelungitoare.

==Date de ieșire==
Dacă datele sunt introduse corect, pe ecran se va afișa: '''"Date de intrare valide."''', apoi fișierul de ieșire <span style=“color: red”> prize.out</span> 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 <span style=“color: red”> N</span> prelungitoare descrise în fișierul de intrare. În cazul în care datele nu respectă restricțiile, se va afișa pe ecran: '''"Date de intrare invalide".'''

==Restricții și precizări==
<span style=“color: red”> 1 ≤ N ≤ 1000</span>
<span style=“color: red”> 1 ≤ numărul de prize dintr-un prelungitor ≤ 5 000 000</span>

==Exemplu:==
<span style=“color: red”> prize.in</span>
: 3
: 3
: 2
: 5

: Date de intrare valide
<span style=“color: red”> prize.out</span>
: 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>
==Explicatie cod:==
Funcția validare(n, nums) primește doi parametri: n de tip întreg și nums o listă de numere întregi. Funcția verifică dacă n este în intervalul [1, 1000] și dacă toate numerele din lista nums sunt în intervalul [0, 5000000]. Dacă toate aceste condiții sunt îndeplinite, funcția returnează True, altfel returnează False.
Funcția prize(n, nums) primește doi parametri: n de tip întreg și nums o listă de numere întregi. Funcția calculează suma tuturor numerelor din lista nums utilizând funcția sum(), iar apoi scade n - 1 din această sumă. Rezultatul obținut este scris în fișierul de ieșire.
În blocul if __name__ == '__main__':, se deschid fișierele de intrare și de ieșire. Se citește fiecare linie din fișierul de intrare și se extrag valorile corespunzătoare. Prima linie este citită ca n, iar liniile următoare sunt convertite într-o listă de numere nums. Se validează datele de intrare utilizând funcția validare(n, nums). Dacă datele de intrare sunt valide, se afișează un mesaj de confirmare, se apelează funcția prize(n, nums) și se scrie rezultatul în fișierul de ieșire. Altfel, se afișează un mesaj de eroare.