PAT(Advanced Level) Practice（with python）——1008 Elevator

Code

N_list = list(map(int,input().split()))
temp = 0
total_time = 0
stay_time = N_list[0]*5
total_time += stay_time
for i in N_list[1:]:
    if i>temp:
        #up:
        total_time += (i-temp)*6
        temp = i
    elif i<temp:
        #down
        total_time += (temp-i)*4
        temp = i
print(total_time)