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

Dacă datele sunt introduse corect, pe ecran se va afișa: "Date de intrare valide.", apoi 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. În cazul în care datele nu respectă restricțiile, se va afișa pe ecran: "Date de intrare invalide".

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

Date de intrare valide

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

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

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.