[Python] 코딩테스트 입문 : 다음에 올 숫자

 

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

 

---

 

문제

 

 

 

---

 

코드

# 등차수열 혹은 등비수열로 매개변수가 주어짐
# 등차수열: 같은 수 더하기
# 등비수열: 같은 수 곱하기
# 마지막 원소 다음 숫자를 찾기
def solution(common):
    # 등차수열인 경우
    val1 = common[-1] - common[-2]
    val2 = common[-2] - common[-3]
    if val1 == val2:
        answer = common[-1] + val1
        return answer
    
    # 등비수열인 경우
    val1 = common[-1] / common[-2]
    val2 = common[-2] / common[-3]
    if val1 == val2:
        answer = common[-1] * val1
        return answer

 

 

 

---