Editing 2644 - Clase

==Cerință==


Într-o școală sunt n clase, fiecare având un număr diferit de elevi. Școală primește m pachete cu cărți, fiecare cu un număr diferit de cărți. Pentru ca o clasa să primească un pachet, numărul elevilor din acea clasa trebuie să fie egal cu numărul cărților din pachet. Să se determine câte clase primesc un pachet de cărți.


==Date de intrare==

Fișierul de intrare '''clase.in''' va avea pe prima linie numărul '''n''', iar pe următoarea linie '''n''' numere naturale ordonate crescător, separate prin spații, reprezentând numărul de elevi din fiecare clasa. Pe următoarea linie numărul '''m''', și pe următoarea linie '''m '''numere, reprezentând numărul de cărți din fiecare pachet.

==Date de ieșire==
Dacă datele sunt introduse corect, pe ecran se va afișa: "Datele de intrare corespund restricțiilor impuse."
Fișierul de ieșire clase.out va conține pe prima linie numărul de clase care primesc pachete cu cărți. În caz contrar, se va afișa pe ecran: "Datele de intrare nu corespund restricțiilor impuse."

==Restricții și precizări==

*'''1 ≤ n, m ≤ 10.000'''
*într-un pachet sunt maximum '''1.000.000.000.000''' cărți.
*într-o clasă sunt maximum '''1.000.000.000.000''' elevi.


==Exemplul 1==
;'''clase.in  '''
:5
:16 17 20 24 30
:4
:78 20 17 74

;'''clase.out '''
:2

;Ieșire
:Datele de intrare corespund restricțiilor impuse.

==Explicație==

Doar clasa cu '''20''' de elevi și cea cu '''17''' elevi vor primi pachete cu cărți.

==Rezolvare==
===Rezolvare ver. 1===
<syntaxhighlight lang="python" line="1" start="1">

def validate_input():
    # Citim datele de intrare
    n = int(input().strip())
    classes = list(map(int, input().strip().split()))
    m = int(input().strip())
    packages = list(map(int, input().strip().split()))

    # Validam inputul
    assert 1 <= n <= 10000, "Numarul de clase trebuie sa fie intre 1 si 10000"
    assert 1 <= m <= 10000, "Numarul de pachete trebuie sa fie intre 1 si 10000"
    for c in classes:
        assert 1 <= c <= 10**12, "Numarul de elevi trebuie sa fie intre 1 si 10^12"
    for p in packages:
        assert 1 <= p <= 10**12, "Numarul de carti trebuie sa fie intre 1 si 10^12"


def count_classes_with_packages(n, classes, m, packages):
    # Initializam un set cu clasele care primesc pachete cu carti
    classes_with_packages = set()

    # Parcurgem toate pachetele cu carti
    for package in packages:
        # Parcurgem toate clasele si verificam daca numarul de elevi
        # dintr-o clasa este egal cu numarul de carti din pachet
        for i in range(n):
            if classes[i] == package:
                classes_with_packages.add(i+1)

    # Returnam numarul de clase care primesc pachete cu carti
    return len(classes_with_packages)


def main():
    # Validam inputul
    validate_input()

    # Citim datele de intrare
    n = int(input().strip())
    classes = list(map(int, input().strip().split()))
    m = int(input().strip())
    packages = list(map(int, input().strip().split()))

    # Determinam numarul de clase care primesc pachete cu carti
    num_classes_with_packages = count_classes_with_packages(n, classes, m, packages)

    # Afisam rezultatul in consola si in fisierul de iesire
    with open("clase.out", "w") as fout:
        fout.write(str(num_classes_with_packages) + "\n")
    print(num_classes_with_packages)


if __name__ == "__main__":
    main()


</syntaxhighlight>

==Explicație==

Mai exact, codul verifică dacă datele de intrare sunt valide prin intermediul funcției validate_input(), care verifică ca numerele de clase și pachete să fie întregi pozitive între 1 și 10000, iar numărul de elevi și cărți să fie între 1 și 10^12.

Apoi, funcția count_classes_with_packages() primește numărul de clase, lista cu numărul de elevi pentru fiecare clasă, numărul de pachete și lista cu numărul de cărți pentru fiecare pachet. Aceasta determină numărul de clase care primesc pachete cu cărți prin parcurgerea tuturor pachetelor și a tuturor claselor și verificarea dacă numărul de elevi dintr-o clasă este egal cu numărul de cărți dintr-un pachet.

În final, codul scrie rezultatul în fișierul de ieșire clase.out și ar trebui să afișeze numărul corect de clase care primesc pachete cu cărți în consolă.