1650 - Acelasi Numar

Sursa: [1]

Cerinţa

Se dă un număr întreg n și alte k numere întregi. Să se afle dacă, adunând toate cele k numere la n se obține o valoare egală cu valoarea inițială a lui n.

Date de intrare

Programul citește de la tastatură numerele n k, iar apoi k numere întregi.

Date de ieșire

Programul va afișa pe ecran textul “DA” dacă numărul final este egal cu cel inițial sau textul “NU” în caz contrar.

Dacă datele sunt introduse corect, programul va rula.

Î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

-1000000 ≤ n ≤ 1000000

2 ≤ k ≤ 10

Exemplul 1

Intrare

25

3

16 -9 3

Ieșire

NU

Rezolvare

#4273
def patrate_perfecte(n):
    patrate = []
    i = 1
    while len(patrate) < n:
        patrat = i * i
        patrate.append(patrat)
        i += 1
    return patrate


def calculeaza(numbers):
    product = 1
    for number in numbers:
        product *= number
    return product


def validare_numar(n):
    if n < 1 or n > 10:
        return False
    return True


if __name__ == '__main__':
    n = int(input("Introduceți numărul n: "))
    if not validare_numar(n):
        print("Datele introduse nu corespund cerintelor.")
    else:
        squares = patrate_perfecte(n)
        product = calculeaza(squares)
        print("Datele introduse corespund cerintelor.")
        print(product)