[프로그래머스 Lv.2][스택/큐] 기능개발 (python)

문제 설명

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

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

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

제한 사항

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

입출력 예progressesspeedsreturn

[93, 30, 55]	[1, 30, 5]	[2, 1]
[95, 90, 99, 99, 80, 99]	[1, 1, 1, 1, 1, 1]	[1, 3, 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개의 기능이 배포됩니다.

✔️ Code

import math

def solution(progresses, speeds):
    answer = []
    stack = []
    for progress, speed in zip(progresses, speeds):
        x = math.ceil((100 - progress) / speed)
        if len(stack) == 0 or (len(stack) > 0 and x > stack[-1]):
            stack.append(x)
            answer.append(1)
        else:
            answer[-1] += 1
            
    return answer

 

✏️ Comment

• zip : 길이가 같은 리스트 등의 요소를 묶어줘서 동시에 for문을 사용할 수 있도록 해주는 내장함수
• ceil : 파이썬의 math 모듈에 포함되어 있는 올림 내장함수 (import. math 선언 필요)

1. 100%가 되어야 기능이 완료된 것이므로 speed에 따라 각각의 progress가 모두 수행되어 기능이 완료되려면 얼마나 걸리는지를 구해서 x라는 변수에 저장한다. 이때, 초반에 (100 - progress) // speed 로 작성했다가 마지막 테스트케이스에서 틀렸었다. 간과하고 있던 부분이 있었는데, 예를 들어 2.3333일이 걸린다 가정했을 때 이는 올림해서 3일이 걸린다고 생각해야 하는데 이 경우를 따져주지 않아서 틀린 것이었다. 따라서 올림을 해주기 위해 math 모듈을 사용하여 math.ceil 내장함수를 사용했다.

2. stack 이라는 배열을 하나 더 만들어서 만약에 다음 기능이 앞의 기능보다 먼저 완성된다면 stack 배열에 집어넣지 않고 answer[-1] 의 값을 + 1 해서 함께 배포될 기능의 갯수를 더해준다. 그리고 만약에 다음 기능이 앞의 기능보다 늦게 완성된다면 stack 배열에 넣어주고 answer 배열에도 그 다음에 배포될 기능의 초기 갯수인 1을 우선 추가해준다. 이 과정을 반복해서 결국엔 마지막에 answer 배열을 출력해주면 원하는 답이 나온다.

progresses와 speeds 배열의 원소들을 동시에 for문을 돌리고 싶어서 찾아보다가 zip이라는 내장함수가 있다는 것을 알았다. 신기방기..  math.ceil을 사용하지 않는 풀이도 있던데 한 번 참고해봐야겠따... 그리고 굳이 stack과 answer 배열 두 개를 쓸 필요가 없을 것 같은데... 나는 바보라서 깔쌈하게 나오지가 않는다. 하핫.

 

반올림, 올림, 내림 함수 알아두기 (참고)

https://jsikim1.tistory.com/223

Python 반올림(round), 올림(ceil), 내림(floor), 소수점 버림(trunc) 하는 방법