Editing 0336 - Expresie4

Sursa: [https://www.pbinfo.ro/probleme/336/expresie4]
== Cerinţa ==
Să se scrie un program care citește numărul natural n și determină suma '''S=1*n+2*(n-1)+3*(n-2)+...+n*1'''.
== Date de intrare ==
Programul citește de la tastatură numărul '''n'''.
== Date de ieșire == 
Programul afișează pe ecran mesajul Rezultatul este S, unde S reprezintă valoarea calculată, în caz contrar acest program va afișa "Datele introduse nu corespund cerințelor".
== Restricţii şi precizări ==
* 0 &lt; '''n''' &les; 100
== Exemplul 1 ==
; Intrare
: 4
; Ieșire
: Rezultatul este 20
<br>
== Exemplul 2 ==
; Intrare
: 0
; Ieșire
: Datele introduse nu corespund cerințelor.
<br>
== Exemplul 3 ==
; Intrare
: 42
; Ieșire
: Rezultatul este 13244
<br>

== Rezolvare == 
<syntaxhighlight lang="python" line>
#0336
def calcul_suma(n):
    suma = 0
    for i in range(1, n+1):
        suma += i*(n-i+1)
    return suma

def validare_n(n):
    if n <= 0:
        print("Datele introduse nu corespund cerintelor.")
        return False
    elif n > 100:
        print("Datele introduse nu corespund cerintelor.")
        return False
    else:
        return True

if __name__ == '__main__':
    n = int(input("Introduceți numărul n: "))

    while not validare_n(n):
        n = int(input("Introduceți un număr valid pentru n: "))

    rezultat = calcul_suma(n)
    print("Rezultatul este", rezultat)


</syntaxhighlight>