[프로그래머스][PYTHON] Lv. 0 다음에 올 숫자

문제 설명

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

프로그래머스

 

내 풀이

def solution(common):
    answer = 0
    if common[1] - common[0] == common[2] - common[1]:
        # 원소 (1-0) = (2-1)이면 등차수열
        answer = common[-1] + (common[1] - common[0])
    else:
        # 등비수열
        answer = common[-1] * (common[1] // common[0])
    return answer

 

단순하게 푼 문제이다. 

원소 간 차가 같으면 등차수열이므로, 마지막 원소에 원소의 차를 더해 다음에 올 원소를 구한다. 

등비수열인 경우 마지막 원소에 앞선 2개의 원소를 나눈 것(몇 배의 곱인지 파악)을 곱하여 다음에 올 원소를 구한다. 

 

느낀점

정답률 64% 문제던데, 생각보다 쉬워서 놀랐다!

뭐지?