[Programmers] 다음에 올 숫자_[-1], [-2]

문제

등차수열 혹은 등비수열 common이 매개변수로 주어질 때, 마지막 원소 다음으로 올 숫자를 return 하도록 solution 함수를 완성해보세요.

 

풀이

def solution(common):
	# 등차수열
    if common[1] - common[0] == common[-1] - common[-2]:
        return common[-1] + (common[1] - common[0])
    # 등비수열
    else:
        return common[-1] * (common[-1] // common[-2])

 

배운 점

양수 인덱스 : 0부터 왼쪽에서 시작
음수 인덱스 : -1부터 오른쪽에서 시작