[Programmers] Stack/Queue - 기능개발 (Python)

출처ㅣ Programmers 코딩테스트 고득점 Kit

Stack/Queue: 기능개발 [Lv2]

문제 설명

프로그래머스 팀에서는 기능 개선 작업을 수행 중입니다. 각 기능은 진도가 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개의 기능이 배포됩니다.

Solution

내 풀이

def solution(progresses, speeds):
    answer = [1]
    size = len(progresses)
    _list = []

    for i in range(size):
        lead_time = (100 - progresses[i]) // speeds[i] + (0 if (100 - progresses[i]) % speeds[i] == 0 else 1)
        _list.append(lead_time)

    for i in range(1, size):
        if _list[i] <= _list[i-1]:
            _list[i] = _list[i-1]
            answer[-1] += 1
        else:
            answer.append(1)

    return answer

_list에 각 기능을 개발하는데에 소요된 기간(lead_time)을 순서대로 넣는다. 개발에 소요된 기간을 구할 때에 100 - 완성도 / 속도가 0으로 나누어 떨어지지 않고 나머지가 있다면, 개발 기간을 올림을 해야하므로, 0 if (100 - progresses[i]) % speeds[i] == 0 else 1을 개발 기간에 더한다.
그 다음 _list에 담긴 값들을 하나씩 비교, 뒤의 값이 앞의 값보다 작으면 answer의 마지막값을 1만큼 올려주고, 뒤의 값이 더 크다면 다른날에 배포해야 하므로 1을 append 해준다.

결과

Best Code

def solution(progresses, speeds):
    Q = []
    for p, s in zip(progresses, speeds):
        if len(Q) == 0 or Q[-1][0] < -((p-100) // s):
            Q.append([-((p-100) // s), 1])
        else:
            Q[-1][1] += 1
    return [q[1] for q in Q]

zip을 사용하여 더 간편하게 풀었다.
Q가 비어있거나 Q[-1][0]이 소요된 기간(-((p-100) // s))보다 작다면 Q에 [소요된 기간, 1]을 추가해준다.
소요된 기간이 -((p-100) // s)인 이유는 math.ceil이나 내가 사용한 삼항연산자를 사용하지 않고 올림을 하기 위함이다. 음수를 s로 나눈 값만 구한 후 이 값에 다시 - 부호를 붙이면 원래 양수의 값을 나눈 후 올림한 것과 같다.
그리고 else문은 Q가 비어있지 않고, Q[-1][0] >= -((p-100) // s)이므로, 현재 순서의 작업이 바로 이전 순서의 작업보다 빨리 끝나는 경우이다.