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

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

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

 

처음 작성한 코드:

a,b,v=map(int,input().split())
day=1
sum=0
while(sum<v):
  sum+=a
  if sum>=v:
    break
  sum-=b
  day+=1
print(day)

물론 정답은 나오지만 while문으로 인한 시간 초과가 일어났다

 

 

정답 코드:

a,b,v=map(int,input().split())
day=(v-b)//(a-b)
if (v-b)%(a-b) ==0:
  print(day)
else:
  print(day+1)

 

풀이 과정:

• 달팽이는 하루에 a-b만큼 올라간다
• 달팽이가 올라가야 하는 길이는 v-b이다. (정상에 올라간 후에는 미끄러지지 않으므로)
• 올라가야 하는 길이 (v-b)를 하루에 올라가는 길이 (a-b)로 나눠준다
• 나머지가 0이 아니라면 하루가 더 필요한 것이기 때문에 +1을 해준다