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

Explicatie

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(n, m, classes, packages):

   # Verificăm ca n și m să fie numere întregi pozitive mai mici sau egale cu 10000
   if not(1 <= n <= 10000) or not(1 <= m <= 10000):
       return False
   
   # Verificăm ca toate clasele să aibă între 1 și 10^12 elevi
   for num_students in classes:
       if not(1 <= num_students <= 10**12):
           return False
   
   # Verificăm ca toate pachetele să aibă între 1 și 10^12 cărți
   for num_books in packages:
       if not(1 <= num_books <= 10**12):
           return False
   
   return True

def count_classes_with_packages(n, classes, m, packages):

   classes_with_packages = 0
   
   for package in packages:
       for num_students in classes:
           if num_students == package:
               classes_with_packages += 1
               break
   
   return classes_with_packages

if __name__ == '__main__':

   # Citim numărul de clase și lista cu numărul de elevi pentru fiecare clasă
   with open('clase.in', 'r') as f:
       n = int(f.readline())
       classes = list(map(int, f.readline().split()))
       
   # Citim numărul de pachete și lista cu numărul de cărți pentru fiecare pachet
   with open('clase.in', 'r') as f:
       m = int(f.readline())
       packages = list(map(int, f.readline().split()))
   
   # Validăm datele de intrare
   if not validate_input(n, m, classes, packages):
       print('Date de intrare invalide!')
       exit()
   
   # Determinăm numărul de clase care primesc pachete cu cărți
   classes_with_packages = count_classes_with_packages(n, classes, m, packages)
   
   # Scriem rezultatul în fișierul de ieșire
   with open('clase.out', 'w') as f:
       f.write(str(classes_with_packages))

</syntaxhighlight>

Explicatie

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