프로그래머스 level0 [다음에 올 숫자] 파이썬

https://school.programmers.co.kr/learn/courses/30/lessons/120924

프로그래머스

엄청 오랜만에 글을 쓰는거 같다. 꾸준히 연습해야지...!

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

먼저 등비수열인지 확인하고, 등비 수열이 아니면 등차가 0인지, 0이 아닌지 확인한다.