2712 - Semne

Sursa: [1]

Cerinţa

Fie un număr n natural nenul, determinați un număr k și o combinație de semne + sau - (mai exact o succesiune x1,x2,…,xk unde xi∈{−1,1}, astfel încat să aibă loc relația: n=x1⋅ 1^2+ x2 ⋅ 2^2 + … + xk ⋅ k^2. Să se afișeze o succesiune de k semne + sau - care să îndeplinească relația de mai sus.

Date de intrare

Programul citește de la tastatură numărul n.

Date de ieșire

Programul va afișa pe ecran, mesajul "Datele introduse corespund cerințelor" și pe o linie nouă programul va afișa pe ecran o succesiune de k semne + sau - care să îndeplinească relația. În caz contrar programul va afișa pe o linie noua mesajul "Datele introduse nu corespund cerintelor."

Restricţii şi precizări

• 1 ⩽ n ⩽ 100

Exemplul 1

Intrare

3

Ieșire

-+--+

Exemplul 2

Intrare

2

Ieșire

---+

Rezolvare

#2712
def validate_n(n):
    if n < 1 or n > 100:
        print("Datele introduse nu corespund cerintelor.")
        return False
    return True


def print_pattern(n):
    if n % 4 == 1:
        print("+", end="")
    elif n % 4 == 2:
        print("---+", end="")
    elif n % 4 == 3:
        print("-+--+", end="")

    for i in range(n // 4):
        print("+--+", end="")


if __name__ == '__main__':
    n = int(input("Introduceti un numar: "))
    if validate_n(n):
        print("Datele introduse corespund cerintelor.")
        print_pattern(n)