Editing 0603 - Vraja (section)

== Rezolvare ==
<syntaxhighlight lang="python" line>
# 0603 - Vraja
def validare(nr_arbori, inaltime_vraja, inaltime_arbori):
    # Daca datele sunt invalide v-om face raise la eroarea ValueError

    conditii = (
        nr_arbori >= 1, nr_arbori <= 1000,
        inaltime_vraja >= 1, inaltime_vraja <= 1000
    )

    if not all(conditii):
        raise ValueError

    for inaltime in inaltime_arbori:
        if (not (inaltime > 0 and inaltime < 1000000)):
            raise ValueError


def vraja(nr_arbori, inaltime_vraja, inaltime_arbori):
    # Vom parcurge arborii în ordine inversă și vom determina pentru fiecare
    # arbore câte vrăji trebuie să i se aplice pentru a deveni cel puțin
    # la fel de înalt ca următorul.

    # Pentru aceasta, vom determina diferența de înălțime dintre cei doi
    # arbori și vom afla numărul de vrăji necesare împărțind diferența la h.

    # Apoi vom actualiza înălțimea copacului curent.
    num_vraji = 0

    for i in range(nr_arbori - 2, -1, -1):
        if inaltime_arbori[i] < inaltime_arbori[i+1]:
            diferenta = inaltime_arbori[i+1] - inaltime_arbori[i]
            nr_vraja = diferenta // inaltime_vraja
            if (nr_vraja * inaltime_vraja < diferenta):
                nr_vraja += 1
            num_vraji += nr_vraja
            inaltime_arbori[i] += nr_vraja * inaltime_vraja

    print(num_vraji)


if __name__ == '__main__':
    # Citim de la tastatura numerele cerute
    nr_arbori = int(input())
    inaltime_vraja = int(input())
    inaltime_arbori = []

    for i in range(nr_arbori):
        inaltime = int(input())

        inaltime_arbori.append(inaltime)

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

        vraja(nr_arbori, inaltime_vraja, inaltime_arbori)

</syntaxhighlight>