[BOJ/백준] 9310 - Arithmetic and Geometric Sums [python]

문제 링크

9310번: Arithmetic and Geometric Sums

문제 풀이

입력값이 등차수열 혹은 등비수열일 경우에 따라 다른 조건으로 계산해준 뒤 출력해준다.

정답 코드

while True :
    n = int(input())
    if n == 0 : break
    a, b, c = map(int, input().split())
    if b - a == c - b : print(n * (2 * a + (n - 1) * (b - a)) // 2)
    else : print(int(a * ((b / a) ** n - 1) / (b / a - 1)))