[코딩테스트] 프로그래머스 기능개발 파이썬(Python)

https://school.programmers.co.kr/learn/courses/30/lessons/42586

프로그래머스

from collections import deque
def solution(progresses, speeds):
    pair = zip(progresses, speeds)
    pair = deque(pair)
    stack = []
    result = []
    while pair:
        progress, speed = pair.popleft()
        remainder = 100 - progress
        x,y = divmod(remainder, speed)
        if y > 0:
            x +=1
        if stack != [] and stack[-1][0] >= x:
            a, b = stack.pop()
            stack.append((a, b+1))
        else:
            stack.append((x, 1))
    result = [i for _, i in stack]
    return result

문제해결을 위하여 stack과 collections 모듈의 deque 자료구조를 활용하였습니다.

zip(), deque()를 사용하여 progresses, speeds 리스트를 쌍으로 묶어서 pair (deque 자료구조)을 만들었습니다.

 

while문을 사용하여 pair가 빌때까지 반복합니다.

popleft()함수를 사용해 pair의 가장 앞에 있는 progress, speed를 추출하고 남은진도 remainder를 계산합니다.

remainder와 speed를 divmod 함수를 사용하여 몫과 나머지를 계산합니다.

x와 y는 각각 몫과 나머지를 의미합니다.

x일간 작업후에 배포가 가능한데 나머지(y)가 존재한다면 x에 1을 더해 줍니다.

예를들어 위의 빨간색 테두리 부분을 확인해보면

두 번째 기능이 30%가 완료되어 남은 진도 remainder은 70%가 됩니다.

이때 70을 주어진 30(speed)으로 몫(x)과 나머지(y)를 계산해주면 x = 2, y=10이 나오게 되는데 이때 2일 지났다고 작업이 완성된것은 아닙니다. 즉 나머지 10이 존재하기 때문에 x에 1을 더해 주는것입니다.

 

그리고 저장할 stack이 비어있지 않거나 stack[-1][0]으로 이전 작업의 배포가 가능한 일자를 확인해보고 stack[-1][0]이 x보다 크거나 같다면

이전작업의 배포가 가능한 일자(a)와 기능을 세는 b를 추출하고 stack에 a와 b+1을 해줍니다.

위의 조건에 부합하지 않다면 stack에 x와 1을 저장줍니다.

 

모든 반복문이 종료되면 result에 기능만 추출하여 반환합니다.