0298 - SumSec

Sursa: 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." și fișierul 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." si fisierul sumsec.out nu va fi generat.

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

Datele sunt introduse corect.

sumsec.out

53

Exemplu 2

Intrare

sumsec.in

8

12 10 15 7 17 10 19 14

Ieșire

Datele nu corespund restricțiilor impuse.

Rezolvare

Rezolvare ver. 1

<syntaxhighlight lang="python" line>

1. 0298 - SumSec

def is_prime(n):

   if n < 2:
       return False
   if n == 2:
       return True
   if n % 2 == 0:
       return False
   for i in range(3, int(n ** 0.5) + 1, 2):
       if n % i == 0:
           return False
   return True

def solve_sumsec(n, a):

   first_prime = -1
   last_prime = -1
   for i in range(n):
       if is_prime(a[i]):
           if first_prime == -1:
               first_prime = i
           last_prime = i
   
   if first_prime == -1 or last_prime == -1:
       return "Datele nu corespund restricțiilor impuse."
   
   sum = 0
   for i in range(first_prime + 1, last_prime):
       sum += a[i]
       
   return sum

def validate_input(n, a):

   if not (1 <= n <= 10000):
       return False
   if any(num < 1 or num > 999999999 for num in a):
       return False
   return True

if __name__ == "__main__":

   n = int(input())
   a = list(map(int, input().split()))

   if not validate_input(n, a):
       print("Datele nu corespund restricțiilor impuse.")
   else:
       print("Datele sunt introduse corect.")
       result = solve_sumsec(n, a)
       with open("sumsec.out", "w") as f:
           f.write(str(result))

</syntaxhighlight>

Explicatie Rezolvare

Funcția is_prime(n) primește ca argument un număr întreg n și verifică dacă acesta este un număr prim. Funcția parcurge toate numerele întregi impare de la 3 la radical din n, verificând dacă n este divizibil cu vreunul dintre acestea. Dacă n este divizibil cu un număr diferit de 1 și de n, atunci n nu este prim și funcția returnează False. În caz contrar, funcția returnează True.

Funcția solve_sumsec(n, a) primește ca argumente un număr întreg n și un vector a cu n elemente și calculează suma elementelor din secvența determinată de primul și ultimul element prim din vectorul dat.

În primul rând, funcția identifică prima și ultima poziție din vectorul a unde se găsesc elemente prime și le reține în variabilele first_prime și last_prime. Dacă nu există cel puțin o astfel de poziție, atunci funcția returnează mesajul "Datele nu corespund restricțiilor impuse."

Dacă există cel puțin o poziție cu un element prim, atunci funcția calculează suma elementelor din secvența determinată de primul și ultimul element prim. Această secvență este reprezentată de toate elementele din vectorul a care se află între pozițiile first_prime și last_prime, cu excepția primului și ultimului element prim din vector.

In blocul if __name__ == "__main__": se citesc datele de intrare din fișierul "sumsec.in", se validează conform restricțiilor problemei, se afișează un mesaj corespunzător și se apelează funcția solve_sumsec() pentru a rezolva problema și a obține rezultatul dorit. Acest rezultat este scris apoi în fișierul "sumsec.out".