백준(Python) 풀이/수학

백준(Python) 4880번 다음수 풀이

개발윗미 2021. 11. 8. 13:49

Python으로 구현한 4880번 다음수 문제 풀이입니다.

 

https://www.acmicpc.net/problem/4880

 

4880번: 다음수

입력은 여러 개의 테스트 케이스로 이루어져 있다. 각 테스트 케이스는 수열의 연속하는 세 정수 a1, a2, a3이 한 줄에 주어진다. (-10,000 < a1, a2, a3 < 10,000) a1, a2, a3은 서로 같지 않다. 입력의 마지막

www.acmicpc.net


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

 

1. while문을 통해 입력받은 세 수가 모두 0일때까지 반복 수행한다.

 

2. 조건문을 통해 두번째 수와 첫번째 수의 차와 세번째 수와 두번째 수의 차가 같을 경우 등차수열에 해당한다.

 

3. 그렇지 않을 경우 등비수열에 해당해 문제에서 요구하는 출력형식에 맞추어 값을 출력한다.