Editing 3687 - Back to the Himalayas (section)

= Exemplul 1: =
Intrare
 9 2 7
 3 2 1 5 3 3 8 2 1
Ieșire
 6 3 3 6 6 6 9 9 9

=== Explicație ===
Hai să vedem de ce <code>ans[1] = 6</code>.

El pornește din primul vârf cu valoare cinetică de <code>49</code>.

De la vârful <code>1</code> la vârful <code>2</code>, valoarea lui crește cu <code>20</code>, deci devine <code>69</code>.

De la vârful <code>2</code> la vârful <code>3</code>, valoarea lui crește cu <code>20</code>, deci devine <code>89</code>.

De la vârful <code>3</code> la vârful <code>4</code>, valoarea lui scade cu <code>80</code>, deci devine <code>9</code>.

De la vârful <code>4</code> la vârful <code>5</code>, valoarea lui crește cu <code>40</code>, deci devine <code>49</code>.

De la vârful <code>5</code> la vârful <code>6</code>, valoarea lui rămâne la <code>49</code>.

De la vârful <code>6</code> la vârful <code>7</code>, valoarea lui scade cu <code>100</code>, deci devine <code>-51</code>, așa că nu mai poate continua drumul.

Astfel, Jany poate ajunge de la vârful <code>1</code> la vârful <code>6</code>.

== Exemplul 2: ==
Intrare
 9 2 40001
 3 2 1 5 3 3 8 2 1
Ieșire
 Datele nu corespund restrictiilor impuse

== Rezolvare == 
<syntaxhighlight lang="python3" line="1">
import sys

# Definirea limitelor și variabilelor
N_MAX = 500000
H = [0] * (N_MAX + 5)
S = [0] * (N_MAX + 5)
ans = [0] * (N_MAX + 5)
L = 0

def get_ans(idx, N, v):
    ret = N + 1
    le, ri = 1, L
    while le <= ri:
        mid = (le + ri) // 2
        if 20 * H[S[mid]] > 20 * H[idx] + v * v:
            ret = S[mid]
            le = mid + 1
        else:
            ri = mid - 1
    return ret - 1

def main():
    global L
    # Citirea datelor de intrare
    try:
        N, dummy, v = input().split()
        N, v = int(N), int(v)
    except ValueError:
        print("Datele introduse nu respectă restricțiile")
        return

    # Verificare restricții pentru N și v
    if not (1 <= N <= 500000 and 1 <= v <= 40000):
        print("Datele introduse nu respectă restricțiile")
        return

    # Citirea înălțimilor de pe o singură linie
    try:
        H[1:N+1] = map(int, input().split())
    except ValueError:
        print("Datele introduse nu respectă restricțiile")
        return

    # Verificarea restricțiilor pentru înălțimi
    if any(h >= 1000000000 for h in H[1:N+1]):
        print("Datele introduse nu respectă restricțiile")
        return

    # Logica principală a programului
    for i in range(N, 0, -1):
        while L > 0 and H[S[L]] <= H[i]:
            L -= 1
        L += 1
        S[L] = i
        ans[i] = get_ans(i, N, v)

    # Afișarea răspunsurilor
    for i in range(1, N + 1):
        print(ans[i], end=" ")
    print()

if __name__ == "__main__":
    main()

</syntaxhighlight>