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

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

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