Editing 0047 - Sumă Pătrate

Sursa: [https://www.pbinfo.ro/probleme/47/sumapatrate]
== Cerinţa ==
Să se scrie un program care citește numărul natural '''n''' și determină suma primelor '''n''' pătrate perfecte nenule.
== Date de intrare ==
Programul citește de la tastatură numărul '''n'''.
== Date de ieșire == 
Programul afișează pe prima linie a ecranului mesajul Rezultatul este '''S''', unde '''S''' reprezintă suma calculată.
== Restricţii şi precizări ==
* 0 &lt; '''n''' &lt; 1000

== Exemplul 1 ==
; Intrare
: 0
; Ieșire
: Datele introduse nu corespund cerințelor.
<br>

== Exemplul 2 ==
; Intrare
: 6
; Ieșire
: Rezultatul este 91
<br>
== Exemplul 3 ==
; Intrare
: 7
; Ieșire
: Rezultatul este 140
<br>

== Rezolvare == 
<syntaxhighlight lang="python" line>
#0047
import math


def validate_input(n):
    if n <= 0 or n >= 1000:
        return False
    else:
        return True


def compute_sum(n):
    sum = 0
    count = 0
    num = 1
    while count < n:
        if math.sqrt(num) % 1 == 0:  # Verifică dacă numărul este pătrat perfect
            sum += num
            count += 1
        num += 1
    return sum


if __name__ == '__main__':
    n = int(input("Introduceti n: "))
    while not validate_input(n):
        print("Datele introduse nu corespund cerintelor.")
        n = int(input("Introduceti n: "))

    sum = compute_sum(n)
    print("Rezultatul este", sum)

</syntaxhighlight>