프로그래머스 디스크 컨트롤러 (python, 파이썬)

• 문제 설명

하드디스크는 한 번에 하나의 작업만 수행할 수 있습니다. 디스크 컨트롤러를 구현하는 방법은 여러 가지가 있습니다. 가장 일반적인 방법은 요청이 들어온 순서대로 처리하는 것입니다.

예를들어

- 0ms 시점에 3ms가 소요되는 A작업 요청 - 1ms 시점에 9ms가 소요되는 B작업 요청 - 2ms 시점에 6ms가 소요되는 C작업 요청

와 같은 요청이 들어왔습니다. 이를 그림으로 표현하면 아래와 같습니다.

한 번에 하나의 요청만을 수행할 수 있기 때문에 각각의 작업을 요청받은 순서대로 처리하면 다음과 같이 처리 됩니다.

- A: 3ms 시점에 작업 완료 (요청에서 종료까지 : 3ms) - B: 1ms부터 대기하다가, 3ms 시점에 작업을 시작해서 12ms 시점에 작업 완료(요청에서 종료까지 : 11ms) - C: 2ms부터 대기하다가, 12ms 시점에 작업을 시작해서 18ms 시점에 작업 완료(요청에서 종료까지 : 16ms)

이 때 각 작업의 요청부터 종료까지 걸린 시간의 평균은 10ms(= (3 + 11 + 16) / 3)가 됩니다.

하지만 A → C → B 순서대로 처리하면

- A: 3ms 시점에 작업 완료(요청에서 종료까지 : 3ms) - C: 2ms부터 대기하다가, 3ms 시점에 작업을 시작해서 9ms 시점에 작업 완료(요청에서 종료까지 : 7ms) - B: 1ms부터 대기하다가, 9ms 시점에 작업을 시작해서 18ms 시점에 작업 완료(요청에서 종료까지 : 17ms)

이렇게 A → C → B의 순서로 처리하면 각 작업의 요청부터 종료까지 걸린 시간의 평균은 9ms(= (3 + 7 + 17) / 3)가 됩니다.

각 작업에 대해 [작업이 요청되는 시점, 작업의 소요시간]을 담은 2차원 배열 jobs가 매개변수로 주어질 때, 작업의 요청부터 종료까지 걸린 시간의 평균을 가장 줄이는 방법으로 처리하면 평균이 얼마가 되는지 return 하도록 solution 함수를 작성해주세요. (단, 소수점 이하의 수는 버립니다)

---

• 제한사항
  • jobs의 길이는 1 이상 500 이하입니다.
  • jobs의 각 행은 하나의 작업에 대한 [작업이 요청되는 시점, 작업의 소요시간] 입니다.
  • 각 작업에 대해 작업이 요청되는 시간은 0 이상 1,000 이하입니다.
  • 각 작업에 대해 작업의 소요시간은 1 이상 1,000 이하입니다.
  • 하드디스크가 작업을 수행하고 있지 않을 때에는 먼저 요청이 들어온 작업부터 처리합니다.

---

• 입/출력

jobs	result
[[0, 3], [1, 9], [2, 6]]	9

---

• 코드

import heapq

def solution(jobs):
    answer = 0
    
    before = -1
    after = 0
    count = 0
    task = []
    
    while count < len(jobs):
        for job in jobs:
            # 앞에 진행되고 있는 작업중에 작업 요청을 받았더라면 해당 task끝나기 전까지 사이의 시간 추가
            if before < job[0] <= after:
                answer += (after-job[0])
                # 중간에 작업 요청을 받았기 때문에 task에 추가시켜준다
                heapq.heappush(task, job[1])
                
        
        # task가 존재할때
        if task:
            # 다른 작업이 진행중이라 작업을 진행하지 못하지만 요청이 들어왔기 때문에 시간은 추가해준다
            answer += task[0]*len(task)
            
            # 시간 time변경
            before = after
            after += heapq.heappop(task)
            
            # 하나의 task가 끝났기 때문에 count를 추가시켜준다
            count += 1
        
        # task가 없을때
        else:
            # 시간만 흐르게 냅둔다
            after += 1
   
    return answer

 

색깔의 선에 맞게 따라가시면 어디 부분을 추가하려는지 나타냈습니다.

한 칸의 두줄로 만들어놓은것은 첫줄은 for문 부분이고 둘째줄은 밑에 task가 있는지 없는지이다.

---

• 출처

programmers.co.kr/learn/courses/30/lessons/42627?language=python3