[백준 - python] 2869번: 달팽이는 올라가고 싶다

<문제 링크>

2869번: 달팽이는 올라가고 싶다

 

<문제 풀이>

A, B, V = map(int, input().split())

if (V - B) % (A - B) == 0 :
    print((V - B) // (A - B))
else :
    print(((V - B) // (A - B)) +1)

2869 입출력 예시

 

*key point: 정상에 도달하면 달팽이가 내려오지 않는다는 점에 유의하기만 하면 if문을 통해 쉽게 계산할 수 있다.

 

 

*유의사항: 해당 문제는 시간 제한이 걸려 있다. 아래와 같은 반복문을 통해 코딩하게 되면 input 값이 커질 경우 시간 초과가 발생할 것이다.

A, B, V = map(int, input().split())
distance = 0
day = 0

while 1:
    day += 1
    distance += A
    if distance >= V:
        print(day)
        break
    distance -= B

따라서 가능한 경우 한번의 계산을 통해 결과가 나올 수 있도록 코딩하는 것이 좋다.

 

*개선 방안1: math 함수 활용을 통해 식을 간단히 할 수 있다.

import math

A, B, V = map(int, input().split())

day = math.ceil((V - A) / (A - B)) + 1
print(day)

 

*개선 방안2: sys모듈의 함수를 활용하여 시간을 단축 할 수 있다.

import sys
import math

A, B, V = map(int, sys.stdin.readline().split())

day = math.ceil((V - A) / (A - B)) + 1
print(day)