[COS Pro 2급, Python] 2차 6번 : 엘리베이터의 총 이동거리구하기

문제

하루 동안 엘리베이터가 멈춘 층이 순서대로 들어있는 배열이 있습니다.

이때, 엘리베이터의 총 이동거리를 구하려 합니다.

단, 층과 층 사이의 거리는 1입니다. 

 

예를 들어 배열에 [1, 2, 5, 4, 2]가 들어있다면, 엘리베이터가 이동한 거리는 7입니다.

 

하루 동안 엘리베이터가 멈춰 선 층이 순서대로 들어있는 배열 floors와 floors의 길이 floors_len이 매개변수로 주어질 때, 엘리베이터의 총 이동 거리를 return 하도록 solution 함수를 작성하려 합니다.

빈칸을 채워 전체 코드를 완성해주세요.

 

 

code

def solution(floors):
	dist = 0
	length = len(floors)
	for i in range(1, length):
		if floors[i] >= floors[i - 1]:
			dist += floors[i] - floors[i-1]
		else:
			dist += floors[i-1] - floors[i]
	return dist
    
    

floors = [1, 2, 5, 4, 2]
ret = solution(floors)

print("solution 함수의 반환 값은", ret, "입니다.")