[BOJ/백준] 4880 - 다음수 [python]

문제 링크

4880번: 다음수

문제 풀이

주어진 입력값으로 등차수열 혹은 등비수열을 판단한 뒤, 네 번째에 올 수를 구한다.

등차수열인지 판단한 뒤, 맞지 않으면 등비수열이므로 쉽게 판단할 수 있다.

정답 코드

while True :
    a_lst = list(map(int, input().split()))
    if a_lst[0] == a_lst[1] == a_lst[2] == 0 : break
    if a_lst[2] - a_lst[1] == a_lst[1] - a_lst[0] :
        print("AP", a_lst[2] + a_lst[2] - a_lst[1])
    else :
        print("GP", int(a_lst[2] * (a_lst[2] / a_lst[1])))