import sys
data = sys.stdin.read().strip().splitlines()
N, spadek = map(int, data[0].split())
droga = list(map(int, data[1].split()))
przed = sum(droga)
# rownamy w prawo
for i in range(1, N):
droga[i] = max(droga[i], droga[i-1]-spadek)
# rownamy w lewo
for i in range(N-2, -1, -1):
droga[i] = max(droga[i], droga[i+1]-spadek)
# i gotowe
print(sum(droga)-przed)
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 | import sys data = sys.stdin.read().strip().splitlines() N, spadek = map(int, data[0].split()) droga = list(map(int, data[1].split())) przed = sum(droga) # rownamy w prawo for i in range(1, N): droga[i] = max(droga[i], droga[i-1]-spadek) # rownamy w lewo for i in range(N-2, -1, -1): droga[i] = max(droga[i], droga[i+1]-spadek) # i gotowe print(sum(droga)-przed) |
English