◾ 디스크 컨트롤러 : 프로그래머스 LEVEL 3
문제
하드디스크는 한 번에 하나의 작업만 수행할 수 있습니다. 디스크 컨트롤러를 구현하는 방법은 여러 가지가 있습니다. 가장 일반적인 방법은 요청이 들어온 순서대로 처리하는 것입니다.
예를들어
- 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 | return |
|---|---|
| [[0, 3], [1, 9], [2, 6]] | 9 |
◾ 풀이
1. 해설
- Heap 자료형을 사용하여 요청된 작업 중 작업 시간이 최소인 경우부터 수행
- 이전 작업 완료 시간, 현재 작업 완료 시간을 통해 그 사이의 요청을 Heap에 추가
- 새로운 작업 완료시 이전 작업 완료 시간, 현재 작업 완료 시간 최신화
- 힙에 작업이 없는 경우 현재 작업 완료 시간을 늘려가며 탐색
- 위 동작 반복
2. 프로그램
- 작업을 (요청 시간, 작업 시간)에 대해 오름차순 정렬
- 모든 작업을 완료할 때까지 반복
1) start, now 사이의 작업 추가
2) 작업 시간이 최소인 경우 수행
3) 작업이 없다면 now + 1 - 모든 작업의 수행 시간 합을 개수로 나눈 몫 반환(소수점 제거)
# 코드
import heapq
def solution(jobs):
answer = 0
# jobs를 요청 시간에 따라 오름차순으로 정렬
jobs.sort(key = lambda x : (x[0], x[1]))
jobs_size = len(jobs) # jobs 개수
jobs_heap = [] # 작업을 담을 (min)heap
# start : 이전 완료 시점
# now : 현재 완료시점
# cnt : 완료한 작업의 수
# idx : jobs 인덱스
start, now, cnt, idx = -1, 0, 0, 0
# 모든 작업을 완료할 때까지 반복
while cnt < jobs_size:
# start와 now 사이의 시간에 요청이 들어올 경우
# jobs_heap에 추가
while idx < jobs_size and start < jobs[idx][0] <= now:
heapq.heappush(jobs_heap, (jobs[idx][1], jobs[idx][0]))
idx += 1
# jobs_heap에 요소가 있는 경우
# 가장 작업 시간이 작은 작업 수행
if jobs_heap:
work, req_time = heapq.heappop(jobs_heap)
start = now
now += work
answer += (now - req_time)
cnt += 1
# jobs_heap에 요소가 없는 경우
# now를 늘려가며 작업 요청 대기
else:
now += 1
# // 연산을 통해 소수점 제거
answer = answer // jobs_size
return answer
후라이드 치킨