0237 - Sume

Sursa: 0237 - Sume

---

Cerinţa

Se dă un şir cu n numere naturale. Să se afişeze suma primilor n termeni din şir, apoi suma primilor n-1 termeni din şir, şi aşa mai departe.

Date de intrare

Fişierul de intrare sume.in conţine pe prima linie numărul n, iar pe a doua linie n numere naturale separate prin spaţii.

Date de ieșire

Dacă datele sunt introduse corect, pe ecran se va afișa: "Datele sunt introduse corect.", apoi, fişierul de ieşire sume.out va conţine n linii, linia i conţinând suma primilor n-i+1 termeni din şir. În cazul în care datele nu respectă restricțiile, se va afișa pe ecran: "Datele nu corespund restricțiilor impuse."'.

Restricţii şi precizări

• 1 ≤ n ≤ 100
• numerele de pe a doua linie a fişierului de intrare vor fi mai mici decât 1000

Exemple

Exemplul 1

sume.in

4

1 7 3 4

Ecran

Datele sunt introduse corect.

sume.out

15

11

8

1

Exemplul 2

sume.in

5

1 7 3 4 1001

Ecran

Datele nu corespund restricțiilor impuse.

Rezolvare

# 0237 sume

def calculeaza_sumele(n, vector):
    with open('sume.out', 'w') as fisier_iesire:
        suma = sum(vector)
        fisier_iesire.write(str(suma) + "\n")
        for i in range(n-1, 0, -1):
            suma -= vector[i]
            fisier_iesire.write(str(suma) + "\n")


def citire_conform_restrictiilor(n, vector):
    if n < 1 or n > 100:
        print("Datele nu corespund restricțiilor impuse.")
        exit()
    for element in vector:
        if element >= 1000:
            print("Datele nu corespund restricțiilor impuse.")
            exit()
    if n != len(vector):
        print("Datele nu corespund restricțiilor impuse.")
        exit()
    print("Datele sunt introduse corect.")


if __name__ == '__main__':
    with open('sume.in', 'r') as fisier_intrare:
        n = int(fisier_intrare.readline().strip())
        vector = list(map(int, fisier_intrare.readline().strip().split()))
    citire_conform_restrictiilor(n, vector)
    calculeaza_sumele(n, vector)

Explicație rezolvare

   Programul de mai sus conține două funcții, funcția calculeaza_sumele(n, vector) și funcția citire_conform_restrictiilor(n, vector), care se vor rula în interiorul main-ului (if __name__ == '__main__' , linia 26) după citirea numărului n (linia 28) și celor n numere pe care le vom pune în șirul „vector” (linia 29), pe care le citim prin deschiderea fișierului sume.in (linia 27).
   După ce am citit toate datele de intrare, se va apela funcția citire_conform_restrictiilor(n, vector) care primește doi parametri: n și vector. Funcția verifică n între 1 și 100 (linia 13), dacă elementele vectorului „vector” sunt mai mici decât 1000 (liniile 16-17) și dacă n este lungimea vectorului „vector” (linia 20). Dacă oricare dintre condiții este încălcată, se va afișa pe ecran mesajul „Datele nu corespund restricțiilor impuse.” și se va ieși din program cu comanda exit(). Dacă toate condițiile sunt respectate, se va afișa mesajul „Datele sunt introduse corect.” (linia 23) și se va continua programul.
   Dacă s-au introdus corect datele, se va apela funcția calculeaza_sumele(n, vector) care primește ca parametrii numărul n și vectorul „vector”, apoi:
-> se deschide fișierul de ieșire 'sume.out' în modul 'w' (write) folosind instrucțiunea with open('sume.out', 'w') as fisier_iesire;
-> se calculează suma tuturor elementelor din vector folosind funcția built-in sum(vector) și se stochează în variabila suma;
-> se scrie suma calculată pe prima linie a fișierului de ieșire utilizând metoda write() a obiectului fisier_iesire, convertind valoarea la tipul șir de caractere folosind str(suma), apoi se adaugă un caracter de linie nouă ("\n") pentru a trece la linia următoare;
-> se parcurge șirul în ordine inversă, începând de la ultimul element până la primul element, utilizând bucla for i in range(n-1, 0, -1). Se folosește un pas negativ -1 pentru a parcurge în ordine inversă. În fiecare pas, se scade valoarea elementului curent din suma pentru a calcula suma primelor n-i+1 termeni din șir, unde i este indicele elementului curent și se scade valoarea elementului curent din suma pentru a calcula suma primelor n-i+1 termeni din șir, unde i este indicele elementului curent, apoi se scrie suma calculată în fiecare pas pe o linie separată în fișierul de ieșire utilizând metoda write() a obiectului fisier_iesire, convertind valoarea la tipul șir de caractere folosind str(suma), apoi se adaugă un caracter de linie nouă ("\n") pentru a trece la linia următoare.