[백준] 2869 달팽이는 올라가고 싶다(브론즈1) - Python

문제

땅 위에 달팽이가 있다. 이 달팽이는 높이가 V미터인 나무 막대를 올라갈 것이다.

달팽이는 낮에 A미터 올라갈 수 있다. 하지만, 밤에 잠을 자는 동안 B미터 미끄러진다. 또, 정상에 올라간 후에는 미끄러지지 않는다.

달팽이가 나무 막대를 모두 올라가려면, 며칠이 걸리는지 구하는 프로그램을 작성하시오.

제한

첫째 줄에 세 정수 A, B, V가 공백으로 구분되어서 주어진다. (1 ≤ B < A ≤ V ≤ 1,000,000,000)

• 시간: 0.15초
• 메모리 제한: 128MB

---

틀린 풀이 (시간 초과)

처음에는 시간 제한을 안 보고 while문을 써서 풀었다가 시간 초과가 떴다.

a, b, v = map(int, input().rstrip().split())

if a >= v:
    print(1)
else:
    days = 0
    climbed = 0

    while climbed < v:
        climbed += a
        days += 1
        
        if climbed >= v:
            break
        else:
            climbed -= b

    print(days)

맞은 풀이

시간 제한을 보니 while문을 돌려서는 절대 맞을 수가 없고 따로 식을 만들어서 풀어야겠다 싶었다. 정상에 올라가면 밤에 미끄러지지 않으니 낮을 x일, 밤을 x-1로 생각해서 계산해서 x이상의 최소 정수가 답이다.

ax - b(x-1) >= v를 계산해서 x >= (v-b)/(a-b)를 올림한 값을 리턴했더니 맞았다. 이번에 올림이 math.ceil이라는 것도 알았다. 분명히 이전에 배운 건데... 까마득하게 잊고 있었다. 뇌야 일해라 제발ㅠ

import math

a, b, v = map(int, input().rstrip().split())

if a >= v:
    print(1)
else:
    print(math.ceil((v-b) / (a-b)))