[프로그래머스] [Python] [PCCE 기출문제] 6번 / 물 부족

1. 문제 설명

ㅇㅇ시에서는 저수지가 하나 있는데, 도시 내에서 사용하는 모든 물은 이 저수지에 저장된 물을 끌어와 사용합니다. 이상 기후로 인해 극심한 가뭄이 예고된 상황에서, 지난 달의 물 사용량과 이번달부터 일정 기간 동안의 월별 물 사용량의 변화를 예측한 값을 이용해 몇 달 뒤 물이 부족해지는지 예측하려고 합니다.
이번달부터의 월별 물 사용량 변화를 예측한 값은 다음과 같이 리스트에 담겨 주어집니다.

• [10, -10, 10, -10, 10, -10, 10, -10, 10, -10]
• 리스트의 각 원소는 해당 월의 물 사용량이 전 달에 비해 몇 %만큼 증가 또는 감소하는지를 나타냅니다.
• 예를 들어, 이번달의 물 사용량 (리스트의 첫 번째 원소)은 지난 달 보다 10% 증가한 값이며, 다음 달(리스트의 두 번째 원소)의 물 사용량은 이번달 사용량에서 10%만큼 감소한 값입니다.

현재 저수지에 저장된 물의 양을 나타내는 정수 storage와 지난 달 물 사용량을 나타내는 정수 usage, 월별 물 사용량이 전 달 대비 어떻게 변하는지 저장된 정수 리스트 change가 주어질 때 몇 달 뒤 물이 부족해지는지 return 하도록 solution 함수를 작성하려 합니다. 코드가 올바르게 작동하도록 한 줄을 수정해 solution 함수를 완성해 주세요. 가뭄이 끝날 때까지 저수지의 물이 남아 있다면 -1을 return합니다.

제한사항

• 1,000 ≤ storage ≤ 1,000,000
• 500 ≤ usage ≤30,000
• 1 ≤ change의 길이 ≤ 30
  • -99 ≤ change[i] ≤ 500
  • change[i]가 양수일 경우 물 사용량은 전 달보다 change[i]%만큼 증가합니다.
  • change[i]가 음수일 경우 물 사용량은 전 달보다 change[i]%만큼 감소합니다.
  • change[i]가 0일 경우 물 사용량은 전 달과 동일합니다.
  • 매달 물 사용량은 소수점 이하를 버린 정수로 계산합니다.

 

2. 출력 예시

 

3. 문제 답안

def solution(storage, usage, change):
    total_usage = 0
    for i in range(len(change)):
        usage = usage * (100 + change[i]) / 100 # 변경 전 usage = total_usage * change[i]/100
        total_usage += usage
        if total_usage > storage:
            return i
    
    return -1