2938 - Albina

Cerința

Într-un stup, pe un fagure de miere infinit, trăiește o albină foarte harnică. Fagurele este alcătuit din celule hexagonale, fiecare celulă având șase vecini. În prima zi albina depune nectar într-o singură celulă. În fiecare din zilele următoarele, albina depune nectar în celulele vecine cu cele în care a fost deja depus nectar. Aflați în câte celule este depus nectar după zile zile.

Date de intrare

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

Date de ieșire

Pe ecran se va afișa mesajul: "Datele de intrare corespund restricțiilor impuse". În următorul rând se va afișa pe ecran numărul celule, reprezentând numărul de celule în care este depus nectar după zile zile.

În cazul în care datele introduse de la tastatură nu îndeplinesc cerințele enunțate, pe ecran se va afișa mesajul "Nu au fost respectate cerintele impuse".

Restricții și precizări

• 1 ⩽ zile ⩽ 1000000

Exemplu 1

Intrare

4

Ieșire

Datele de intrare corespund restricțiilor impuse

37

Exemplu 2

Intrare

0

Ieșire

Nu au fost respectate cerintele impuse

Rezolvare

# 2938 - Albina
def validare(zile):
    # Daca datele sunt invalide v-om face raise la eroarea ValueError

    if (not (zile >= 1 and zile <= 1000000)):
        raise ValueError


def albina(zile):
    # Numărul de celule în care se depune nectar într-o anumită zi depinde de 
    # numărul de celule în care s-a depus nectar în ziua precedentă.

    # Pentru a afla numarul de celule in care au fost depus nectar
    # trebuie sa inmultim 3 cu zile si (zile - 1)
    # dupa care adunam cu 1
    celule = 1 + 3 * zile * (zile - 1)

    print(celule)


if __name__ == '__main__':
    # Citim de la tastatura numerele cerute
    zile = int(input())

    # Verificam daca datele introduse sunt valide
    # daca nu afisam mesajul corespunsador
    try:
        validare(zile)
    except ValueError:
        print("Nu au fost respectate cerintele impuse")
    else:
        print("Datele de intrare corespund restricțiilor impuse")

        albina(zile)