[프로그래머스] 기능 개발 level2 - python

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

프로그래머스

 

 

 

문제 설명

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

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

 

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

 

 

 

 

 

제한사항

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

 

 

 

 

 

 

입출력 예

 

 

---

이 문제는 총 2가지의 단계로 나누어 접근하였다. 먼저 각 작업이 완료되기 위한 일수를 포함하는 리스트가 필요했으며, 그 후에 구한 리스트를 바탕으로 return값을 출력하였습니다.

 

첫 번째 입출력 예를 보자면 각 작업의 진도는 는 [93, 30, 55] 이고, 각 작업의 개발 속도는 [1, 30, 5] 이다.

 

그렇다면 각 작업이 완료되기 위해 필요한 일수는 [93, 30, 55] → [7, 3, 9] 가 된다.

 

마지막으로 작업을 시작하게 되면 7일이 걸리는 첫 번째 작업이 완료되는 동시에 7일보다 더 적게 걸리는 3일치의 두 번째 작업이 같이 완료된다. 그리고 7일보다 더 오래 걸리는 9일치의 작업은 그 후에 완료되기 때문에 출력값은 [2, 1] 이 된다.

 

 

 

 

 

 

 

 

 

 

stack 사용

import math
def solution(progresses, speeds):
    # 각 작업이 완료되기 위한 일수
    stack = []
    for prog, spe in zip(progresses, speeds):
        day = (100 - prog)/spe
        stack.append(math.ceil(day))
    
    
    # stack ==> ##############
    # [7, 3, 9]
    # [5, 10, 1, 1, 20, 1]
    ##########################
    

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

이 문제를 풀기 위해 먼저 각 작업이 완료되기 위한 일수를 포함하는 리스트를 구해주었다. 그 후에 반복문을 사용하여 리스트의 첫 번째 값을 꺼낸 후 그 값이 리스트의 첫 번째 값보다 큰 경우 cnt를 1씩 늘려감과 동시에 0번째인 값을 pop을 해주었다. 

 

 

 

 

 

 

다음 코드는 위의 방식과 접근하는 방식은 동일하나 푸는 방식에 차이가 있다.

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]

 

첫 번째 코드는 2번의 단계로 나누어 문제를 해결하였다면 이번 코드는 그 2번의 단계를 리스트 안에 리스트를 넣어 한번에 해결하였다. 

 

똑같이 첫 번째 입출력 예시인 [93, 30, 55], [1, 30, 5] 를 넣게 되면 [ [7, 2], [9, 1] ] 이라는 Q리스트가 나온다. 애초에 처음 리스트를 생성할 때 같이 완료되는 작업의 개수를 같이 넣어주어 코드가 훨씬 간단해지는 것을 볼 수 있다.