[프로그래머스] [python] 스택/큐 - 기능개발

문제 설명

프로그래머스 팀에서는 기능 개선 작업을 수행 중입니다. 각 기능은 진도가 100%일 때 서비스에 반영할 수 있습니다.

또, 각 기능의 개발속도는 모두 다르기 때문에 뒤에 있는 기능이 앞에 있는 기능보다 먼저 개발될 수 있고, 이때 뒤에 있는 기능은 앞에 있는 기능이 배포될 때 함께 배포됩니다.

먼저 배포되어야 하는 순서대로 작업의 진도가 적힌 정수 배열 progresses와 각 작업의 개발 속도가 적힌 정수 배열 speeds가 주어질 때 각 배포마다 몇 개의 기능이 배포되는지를 return 하도록 solution 함수를 완성하세요.

제한 사항

• 작업의 개수(progresses, speeds배열의 길이)는 100개 이하입니다.
• 작업 진도는 100 미만의 자연수입니다.
• 작업 속도는 100 이하의 자연수입니다.
• 배포는 하루에 한 번만 할 수 있으며, 하루의 끝에 이루어진다고 가정합니다. 예를 들어 진도율이 95%인 작업의 개발 속도가 하루에 4%라면 배포는 2일 뒤에 이루어집니다.

입출력 예 설명

입출력 예 #1
첫 번째 기능은 93% 완료되어 있고 하루에 1%씩 작업이 가능하므로 7일간 작업 후 배포가 가능합니다.
두 번째 기능은 30%가 완료되어 있고 하루에 30%씩 작업이 가능하므로 3일간 작업 후 배포가 가능합니다. 하지만 이전 첫 번째 기능이 아직 완성된 상태가 아니기 때문에 첫 번째 기능이 배포되는 7일째 배포됩니다.
세 번째 기능은 55%가 완료되어 있고 하루에 5%씩 작업이 가능하므로 9일간 작업 후 배포가 가능합니다.

따라서 7일째에 2개의 기능, 9일째에 1개의 기능이 배포됩니다.

입출력 예 #2
모든 기능이 하루에 1%씩 작업이 가능하므로, 작업이 끝나기까지 남은 일수는 각각 5일, 10일, 1일, 1일, 20일, 1일입니다. 어떤 기능이 먼저 완성되었더라도 앞에 있는 모든 기능이 완성되지 않으면 배포가 불가능합니다.

따라서 5일째에 1개의 기능, 10일째에 3개의 기능, 20일째에 2개의 기능이 배포됩니다.

문제 풀이

import math

def solution(progresses, speeds):
    answer = []
    # 남은 일수 계산
    days = []
    for i in range(len(progresses)):
        day = (100 - progresses[i]) / speeds[i]
        days.append(math.ceil(day))
    
    # 배포 계산
    day = days[0]
    answer.append(1)
    answer_index = 0
    for i in range(1,len(days)):
        if days[i] > day:
            answer.append(1)
            answer_index += 1
            day = days[i]
        else:
            answer[answer_index] += 1
    return answer

위의 입출력 예에서 설명된 남은 날짜를 계산하기 위해 progresses를 순회하며 걸리는 날짜를 계산한다.

math.ceil 함수를 사용하여 나누어 떨어지지 않는 날짜는 올림으로 계산하였다.

입출력 예를 통한 설명

입출력 예시로 더 자세하게 설명해 보겠다.

progresses = [93, 30, 55], speeds = [1, 30, 5] 일때 days = [7,3,9] 로 계산된다.

days를 순회하며 각 배포마다 몇 개의 기능이 배포되는지를 계산해야 했다.

생각보다 간단하게, days 배열에서 다음 요소가 더 작다면 현재 배포 단계에 +1을 해주고, 다음 요소가 더 크다면 다음 배포 단계로 넘어가는 것이다.

    day = days[0] # 첫 번째 원소
    answer.append(1) # 첫 번째 배포 단계
    answer_index = 0 # 현재 배포 단계를 알고있기 위한 인덱스
    
    for i in range(1,len(days)):
    	# 다음 day가 더 길다면 다음 배포 단계로 넘어가기
        if days[i] > day: 
            answer.append(1)
            answer_index += 1
            # 현재 day 갱신
            day = days[i]
        # 현재 day가 더 적다면 현재 배포 단계에 기능 추가
        else:
            answer[answer_index] += 1

내 코드 개선

이 코드의 단점이라고 한다면, progresses 배열을 순회한 후에 다시 계산된 days 배열을 또 순회해야 한다는 것이다.

import math

def solution(progresses, speeds):
    answer, days = [], []
    answer_index, deploy_day = -1, -1
    
    for i in range(len(progresses)):
        # 남은 일수 계산
        day = math.ceil((100 - progresses[i]) / speeds[i])
        days.append(day)
        
        # 배포 단계 계산
        if day > deploy_day:
            answer.append(1)
            answer_index += 1
            deploy_day = day
        else:
            answer[answer_index] += 1
        
    return answer

배열을 한번만 순회할때 모두 처리하도록 개선한 코드이다. 

위의 코드는 첫번째 day를 지정하고 시작했지만, 배열을 한번만 순회하도록 하기 위해

현재 배포되는 데 걸리는 시간 = deploy_day = -1 로 설정하고 시작하였다.