[구현/수학] 백준 4880 다음수 - 파이썬(Python)

 

[ Contents ]

 

 

1. 문제 (링크 참조)

4880번: 다음수

 

 

2. 문제 풀이

 주어진 수열이 등차인지, 등비인지 판별하고 그 다음에 올 항을 구하는 문제입니다.

 

 

3. 코드

import sys
input = sys.stdin.readline

while True:
    a, b, c = map(int, input().split())
    if a == b == c == 0:
        break
    if b-a == c-b:
        print(f"AP {c + c-b}")
    else:
        print(f"GP {c * (c//b)}")

 먼저 등차인지, 등비인지부터 판별합니다.

 그리고 해당 수열에 따라 마지막 항에 등차 더하거나, 등비를 곱해서 다음 항을 구합니다.