[Python]스택/큐: 기능개발

코드싸개·2021년 1월 14일

programmers

목록 보기

7/20

문제 설명

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

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

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

제한 사항

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

입출력 예

progresses	speeds	return
[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개의 기능이 배포됩니다.

내 코드

import math
def solution(progresses, speeds):
    answer = []
    days = []
    for i in range(len(speeds)):
        remain_days = math.ceil((100 - progresses[i]) / speeds[i])
        days.append(remain_days)

    cnt = 1
    while days:
        if len(days) == 1:
            answer.append(cnt)
            break
        if days[0] >= days[1]:
            days.pop(1)
            cnt += 1
        else:
            days.pop(0)
            answer.append(cnt)
            cnt = 1
    return answer

코딩 테스트 문제를 풀면서 처음으로 import를 사용해본 것 같다.

각 기능들의 개발까지 얼마의 시간이 필요할 지 구해야 할 것 같아 days라는 리스트를 만들고, 100에서 개발진행률을 빼고 그 값을 speed로 나눈 후 math의 ceil을 이용해 올림 처리하였다.
cnt를 1로 설정해두고 days로 while문을 돌리면서 선행하는 개발이 같거나 크면 뒤에 개발을 pop하고 cnt를 1씩 증가시키며, 반대로 선행하는 개발이 작으면 더 같이 배포될 필요가 없으므로 선행하는 개발을 pop하고 answer에 cnt 값을 넣고 cnt 값을 다시 리셋해주었다.

정확성 테스트
테스트 1 〉 통과 (0.02ms, 10.2MB)
테스트 2 〉 통과 (0.07ms, 10.1MB)
테스트 3 〉 통과 (0.05ms, 10.2MB)
테스트 4 〉 통과 (0.03ms, 10.2MB)
테스트 5 〉 통과 (0.01ms, 10.2MB)
테스트 6 〉 통과 (0.01ms, 10.2MB)
테스트 7 〉 통과 (0.05ms, 10.1MB)
테스트 8 〉 통과 (0.02ms, 10.2MB)
테스트 9 〉 통과 (0.04ms, 10.2MB)
테스트 10 〉 통과 (0.05ms, 10.1MB)
테스트 11 〉 통과 (0.01ms, 10MB)

채점 결과
정확성: 100.0
합계: 100.0 / 100.0

다른 사람의 생각

zip을 이용한 코드

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으로 progresses와 speeds를 묶어 for 문을 통해 각각 값을 뽑아내며, 걸리는 시간과 1로 만들어진 리스트 Q에 넣는다.
ceil을 사용하지 않기 위해 -((p-100)//s) 를 사용했는데 이는 아래와 같다.
- (p-100) -> 음수
- (p-100)//s -> 내림한 음수, 이 때 절댓값이 커진다.
- -((p-100)//s) -> 올림한 양수
q[1] 에는 같이 배포될 갯수가 있으므로 이를 리스트 컴프리헨션으로 뽑아낸다.

예외 처리를 이용한 코드

from math import ceil

def solution(progresses, speeds):
    daysLeft = list(map(lambda x: (ceil((100 - progresses[x]) / speeds[x])), range(len(progresses))))
    count = 1
    retList = []

    for i in range(len(daysLeft)):
        try:
            if daysLeft[i] < daysLeft[i + 1]:
                retList.append(count)
                count = 1
            else:
                daysLeft[i + 1] = daysLeft[i]
                count += 1
        except IndexError:
            retList.append(count)

    return retList