[BaekJoon] 4880번 : 다음수 (Python)

코딩 1일 1문제! 오늘의 문제는 백준의 다음수 입니다.

4880번: 다음수

👨🏻‍💻 문제 풀이

입력 받은 숫자 리스트가

등차 수열이면 AP 와 함께 다음 수

등비 수열이면 GP 와 함께 다음 수 를 출력합니다.

 

각 숫자 사이의 비율이 같으면 등차 수열 

그렇지 않으면 등비 수열로 판단하고

 

등차 수열일 경우 

세번째 수 + 두번째 수 - 첫번째 수

등비 수열일 경우

세번째 수 + 두번째 수 // 첫번째 수

를 정답으로 합니다.

👨🏻‍💻 코드 ( Solution )

def daumsu(number_list):
    if (number_list[1] - number_list[0]) == (number_list[2] - number_list[1]):
        answer = f"AP {number_list[2] + number_list[1] - number_list[0]}"
    else:
        answer = f"GP {number_list[2] * (number_list[1] // number_list[0])}"
        
    return answer


if __name__ == "__main__":
    while True:
        number_list = list(map(int, input().split()))
        
        if number_list == [0, 0, 0]:
            break
            
        print(daumsu(number_list))

GitHub - SOMJANG/CODINGTEST_PRACTICE: 1일 1문제 since 2020.02.07