[Python] 프로그래머스. 디스크 컨트롤러 (Level 3)

https://programmers.co.kr/learn/courses/30/lessons/42627

코딩테스트 연습 - 디스크 컨트롤러 | 프로그래머스

heapq 라이브러리를 사용하고, 프로세스가 queue에 들어올 수 있는 경우 / 여러 프로세스가 queue에 들어온 경우 시간을 어떻게 계산할 것인지가 꽤 고민스러운 문제.

 

 

	import heapq
	def solution(jobs):
	n = len(jobs)
	time, end, queue = 0, -1, []
	# 처리한 프로세스 개수
	count = 0
	answer = 0
	while count < n:
	for i in jobs:
	# i[0] = 프로세스 입력 시간, i[1] = 프로세스 끝날 때까지 걸리는 시간
	if end <i[0] <= time:
	# 현재 시간 기준, 프로세스가 queue에 들어가서 얼마나 기다렸는지.
	answer += (time - i[0])
	heapq.heappush(queue, i[1])
	if len(queue) > 0:
	# 가장 빨리 끝나는 프로세스가 끝날 때까지는 queue에 있는 프로세스 전부 대기시간이므로 값을 추가한다.
	answer += len(queue) * queue[0]
	# 끝난 시간
	end = time
	# 가장 빨리 끝나는 프로세스가 걸린 시간을 더해준다
	time += heapq.heappop(queue)
	# 프로세스가 끝났으므로 count + 1 해준다.
	count += 1
	else:
	# queue에 아직 아무것도 없으므로 시간을 +1 해준다.
	time += 1
	return (int(answer / n))

view raw [프로그래머스] 디스크 컨트롤러.py hosted with ❤ by GitHub