다음에 올 숫자.python

문제 설명

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

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

입출력 예
common	        result
[1, 2, 3, 4]	5
[2, 4, 8]	    16

풀이

처음 변수 선언시 0이 들어가서 런타임 에러가 발생하였다. 분모가 0이기에
차라리 if 문안에 넣어서 생략해 버리면 좋았다.

def solution(common):
    a = common[-1]-common[-2]
    if common[-2] == 0:  
        b = None
    else:
        b = common[-1] / common[-2]  
    if (common[1] - common[0]) == a :
        return common[-1] + a
    else :
        return common[-1] * b


def solution(common):
    a = common[-1]-common[-2]
     
    if (common[1] - common[0]) == a :
        return common[-1] + a
    else :
        return common[-1] * (common[-1] // common[-2])