[Programmers][Python] 다음에 올 숫자

문제 링크:https://school.programmers.co.kr/learn/courses/30/lessons/120924

프로그래머스

 

처음 작성한 코드

• 문제에서 등차수열과 등비수열 둘 중 하나임을 명시했으므로, 등차수열에 대한 조건식만 작성해도 좋다.
• 등차수열임을 확인하기 위해 0과 1번째 요소의 차와 1과 2번째 요소의 차를 비교해 준다.
  • 맞다면 마지막 요소와 차이 값을 더해주어 반환한다.
  • 아니다면 마지막 요소와 차이 값을 곱해주어 반환한다.

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

# 입출력 예시
print(solution([1, 2, 3, 4]))
print(solution([2, 4, 8]))

간단한 문제이고, 만약 조건식을 추가해 준다 해도 어렵지 않다.

 

다른 코드

사소한 차이는 있으나 크게 다른 코드, 혹은 더 좋은 코드를 발견하지 못했다.