import sys
def koparka():
l1 = "4 2"
l2 = "7 3 0 2"
l1 = "4 0"
l2 = "0 10"
l1 = sys.stdin.readline()
l2 = sys.stdin.readline()
l1 = l1.split()
gaps = int(l1[1])
budowa = l2.split()
not_plot_relevant_enough = 0
for i in range(len(budowa)):
budowa[i] = int(budowa[i])
while True:
max_ = 0
for i in range(1,len(budowa)):
if budowa[i] > budowa[max_]:
max_ = i
#print("m ", max_, " ", budowa[max_], " piach ", not_plot_relevant_enough)
if budowa[max_] == -1:
break
sprawdz = max_ - 1
mini = budowa[max_] - gaps
if sprawdz >= 0 and budowa[sprawdz] != -1 and budowa[sprawdz] < mini:
betoniarka = mini - budowa[sprawdz]
not_plot_relevant_enough += betoniarka
budowa[sprawdz] = mini
sprawdz = max_ + 1
if sprawdz < len(budowa) and budowa[sprawdz] != -1 and budowa[sprawdz] < mini:
betoniarka = mini - budowa[sprawdz]
not_plot_relevant_enough += betoniarka
budowa[sprawdz] = mini
budowa[max_] = -1
print(not_plot_relevant_enough)
koparka()
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 | import sys def koparka(): l1 = "4 2" l2 = "7 3 0 2" l1 = "4 0" l2 = "0 10" l1 = sys.stdin.readline() l2 = sys.stdin.readline() l1 = l1.split() gaps = int(l1[1]) budowa = l2.split() not_plot_relevant_enough = 0 for i in range(len(budowa)): budowa[i] = int(budowa[i]) while True: max_ = 0 for i in range(1,len(budowa)): if budowa[i] > budowa[max_]: max_ = i #print("m ", max_, " ", budowa[max_], " piach ", not_plot_relevant_enough) if budowa[max_] == -1: break sprawdz = max_ - 1 mini = budowa[max_] - gaps if sprawdz >= 0 and budowa[sprawdz] != -1 and budowa[sprawdz] < mini: betoniarka = mini - budowa[sprawdz] not_plot_relevant_enough += betoniarka budowa[sprawdz] = mini sprawdz = max_ + 1 if sprawdz < len(budowa) and budowa[sprawdz] != -1 and budowa[sprawdz] < mini: betoniarka = mini - budowa[sprawdz] not_plot_relevant_enough += betoniarka budowa[sprawdz] = mini budowa[max_] = -1 print(not_plot_relevant_enough) koparka() |
English