다음에 올 숫자 - Python

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


# Test Cases
print(solution([1, 2, 3, 4]))
print("=" * 50)
print(solution([2, 4, 8]))
print("=" * 50)

  위와 같이 풀었습니다.

 

---

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

  가독성 좋게 푼 풀이를 발견했습니다.

 

---

• 프로그래머스: https://school.programmers.co.kr/learn/courses/30/lessons/120924

프로그래머스