프로그래머스 5일차 - 다음에 올 숫자

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

제한사항
2 < common의 길이 < 1,000
-1,000 < common의 원소 < 2,000
common의 원소는 모두 정수입니다.
등차수열 혹은 등비수열이 아닌 경우는 없습니다.
등비수열인 경우 공비는 0이 아닌 정수입니다.

입출력
예 #1[1, 2, 3, 4]는 공차가 1인 등차수열이므로 다음에 올 수는 5이다.
예 #2[2, 4, 8]은 공비가 2인 등비수열이므로 다음에 올 수는 16이다.

해설

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

 

간단하다. 등비 or 등차 두가지 경우만 나온다.

 

리스트 요소 수는 무조건 3개 이상이니까. 1,0 번째와 2,1번째 인덱스 요소를 비교하면 된다.

그리고 마지막 리스트 요소 -1번째에다가 더하든 곱해주든 하면 

return 결과가 나온다.