Lv.0 다음에 올 숫자

문제 설명

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

입출력

코드 풀이

def solution(common):
    answer = 0
    a,b,c = common[:3]
    if (b-a) == (c-b):
        return common[-1]+(b-a)
    else:
        return common[-1] * (b//a)
    return answer

• (b-a)와 (c-b)를 비교하여 두 값이 같으면 등차수열로 판별 / 두 값이 다르면 등비수열로 판별

출처: 프로그래머스 코딩테스트 연습