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==
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.

==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

<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 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.