프로그래머스 다음에 올 숫자

문제

def solution(common):

    length = len(common)
    flag = 0
    for i in range(length - 1, 0, -1):

        gap = common[i] - common[i - 1]

        if flag == gap:
            return common[length - 1] + flag

        flag = gap

    gap = common[length - 1] // common[length - 2]
    return common[length - 1] * gap

다른 사람 풀이

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

2 < common의 길이 < 1,000 이므로 [0], [1], [2]는 무조건 존재하므로 이를 이용해서 풀었다.