0298 - SumSec

Cerinţa

Se dă un şir format din n elemente, numere naturale. Calculaţi suma elementelor din secvenţa determinată de primul şi ultimul element prim.

Date de intrare

Fişierul de intrare sumsec.in conţine pe prima linie numărul n; urmează cele n elemente ale vectorului, dispuse pe mai multe linii şi separate prin spaţii.

Date de ieșire

Dacă datele sunt introduse corect, pe ecran se va afișa: '"Datele sunt introduse corect.", apoi pe un rând nou sumsec.out va conţine pe prima linie numărul S, reprezentând suma calculată, reprezentând valoarea cerută. În cazul contrar, se va afișa pe ecran: "Datele nu corespund restricțiilor impuse.".

Restricţii şi precizări

1 ≤ n ≤ 1000
elementele şirului vor avea cel mult 9 cifre
şirul va conţine cel puţin un element prim

Exemplu 1

Intrare: sumsec.in; 8; 12 10 15 7 17 10 19 14
Ieșire: sumsec.out; Datele sunt introduse corect.; 53

Exemplu 2

Intrare: sumsec.in; 8; 12 10 15 7 17 10 19 14
Ieșire: sumsec.out; Datele nu corespund restricțiilor impuse.; 21

Rezolvare

Rezolvare ver. 1

0298 - SumSec

def citire():

   n, k = map(int, input().split())
   sir = list(map(int, input().split()))
   return n, k, sir

def rezolvare(n, k, sir):

   if k < 1 or k > n:
       return None  # Returnam None daca k nu respecta conditiile
   secventa = []
   secvente = 0

   for i in range(n):
       if sir[i] % 2 == 0:
           continue  # Ignoram elementele pare
       if len(secventa) == 0 or len(secventa[-1]) == k:
           secventa.append([sir[i]])  # Cream o secventa noua
           secvente += 1
       else:
           secventa[-1].append(sir[i])  # Adaugam elementul la ultima secventa

   return secvente

if __name__ == '__main__':

   try:
       n, k, sir = citire()
       rez = rezolvare(n, k, sir)
       if rez is not None:
           print("Datele sunt introduse corect.")
           print(rez)
       else:
           print("Datele nu corespund restricțiilor impuse.")
   except ValueError:
       print("Datele nu corespund formatului cerut.")  # In cazul in care input-ul nu este corect din punct de vedere al formatului

</syntaxhighlight>

Explicatie Rezolvare

Citim datele de intrare si construim vectorul de numere. Cautam primul si ultimul element prim din vector. Determinam secventa de numere intre primul si ultimul element prim. Calculam suma elementelor din secventa determinata la pasul anterior. Functia citire_date citeste datele de intrare din fisierul "sumsec.in" si returneaza vectorul de numere v. Functia este_prim verifica daca un numar este prim sau nu. In functia rezolvare cautam primul si ultimul element prim din vectorul v, si apoi calculam suma elementelor din secventa determinata de acestea. Observam ca secventa incepe de la primul+1 si se termina inaintea lui ultimul, deoarece primul si ultimul element prim nu fac parte din secventa. Functia validare compara valoarea calculata cu cea din fisierul "sumsec.out" si returneaza True daca acestea coincid, si False altfel. Am adăugat o verificare suplimentară în funcția rezolvare() pentru a returna None dacă k nu respectă restricțiile cerute. Apoi, în blocul if __name__ == '__main__', verificăm dacă rezultatul returnat este diferit de None și afișăm mesajul corespunzător. Am introdus, de asemenea, un bloc try-except pentru a gestiona cazul în care valorile de intrare nu sunt în formatul așteptat. In if __name__ == "__main__": apelam cele trei functii in ordinea prezentata mai sus, si afisam un mesaj corespunzator, in functie de rezultatul validarii.