문제

문제 설명

운영체제의 역할 중 하나는 컴퓨터 시스템의 자원을 효율적으로 관리하는 것입니다. 이 문제에서는 운영체제가 다음 규칙에 따라 프로세스를 관리할 경우 특정 프로세스가 몇 번째로 실행되는지 알아내면 됩니다.

1. 실행 대기 큐(Queue)에서 대기중인 프로세스 하나를 꺼냅니다.
2. 큐에 대기중인 프로세스 중 우선순위가 더 높은 프로세스가 있다면 방금 꺼낸 프로세스를 다시 큐에 넣습니다.
3. 만약 그런 프로세스가 없다면 방금 꺼낸 프로세스를 실행합니다.
  3.1 한 번 실행한 프로세스는 다시 큐에 넣지 않고 그대로 종료됩니다.
예를 들어 프로세스 4개 [A, B, C, D]가 순서대로 실행 대기 큐에 들어있고, 우선순위가 [2, 1, 3, 2]라면 [C, D, A, B] 순으로 실행하게 됩니다.

현재 실행 대기 큐(Queue)에 있는 프로세스의 중요도가 순서대로 담긴 배열 priorities와, 몇 번째로 실행되는지 알고싶은 프로세스의 위치를 알려주는 location이 매개변수로 주어질 때, 해당 프로세스가 몇 번째로 실행되는지 return 하도록 solution 함수를 작성해주세요.

제한사항

priorities의 길이는 1 이상 100 이하입니다.
priorities의 원소는 1 이상 9 이하의 정수입니다.
priorities의 원소는 우선순위를 나타내며 숫자가 클 수록 우선순위가 높습니다.
location은 0 이상 (대기 큐에 있는 프로세스 수 - 1) 이하의 값을 가집니다.
priorities의 가장 앞에 있으면 0, 두 번째에 있으면 1 … 과 같이 표현합니다.

입출력 예

priorities	location	return
[2, 1, 3, 2]	2	1
[1, 1, 9, 1, 1, 1]	0	5

입출력 예 설명

예제 #1

문제에 나온 예와 같습니다.

예제 #2

6개의 프로세스 [A, B, C, D, E, F]가 대기 큐에 있고 중요도가 [1, 1, 9, 1, 1, 1] 이므로 [C, D, E, F, A, B] 순으로 실행됩니다. 따라서 A는 5번째로 실행됩니다.

풀이

문제가 머리로는 이해가 가는데 아직 큐를 다루는 데 미숙하다보니 코드로 옮기는 과정이 생각보다 어려웠다ㅜㅜ 그래서 다른 사람의 풀이를 참고해보았다.

1. [(index, value), (index, value)] 모양 만들기

방법 1. for문 사용

queue = [] 

for i in range(len(priorities)):  # 리스트의 길이만큼 반복하면서
    queue.append((i, priorities[i]))  # (인덱스, 값) 튜플을 추가

방법 2. enumerate 사용

enumerate()

반복 가능한(iterable) 객체(리스트, 튜플 등)를 순회할 때 인덱스와 요소를 동시에 가져오는 함수

queue = []

for i, p in enumerate(priorities):  # enumerate() 사용
    queue.append((i, p))  # (인덱스, 값) 튜플 추가

방법 3. 리스트 컴프리헨션 + enumerate 사용

queue =  [(i,p) for i,p in enumerate(priorities)] # (index, priorities)

2. 현재 요소의 우선순위와 queue 안의 우선순위를 비교

우선 cur = queue.pop(0)해서 큐의 가장 첫 번째 요소를 pop한다.

if any(cur[1] < q[1] for q in queue):
    queue.append(cur)

현재 요소의 우선순위보다 queue에 있는 어떤 요소의 우선순위가 더 큰 게 있다면 queue에 현재 요소를 새로 넣는다.(append)

=> any(iterable 객체): 하나라도 True라면 True 반환

3. 현재 요소의 우선순위가 가장 크다면

else: # 현재 요소의 우선순위가 가장 크다면 pop으로 인해 개수 하나가 빠지는 것이므로
    answer += 1 # answer의 index 늘리고
    if cur[0] == location: # 만약 현재 index가 location과 같다면 답
        return answer

최종 코드

def solution(priorities, location):
    answer = 0
    queue = []

    for i, p in enumerate(priorities):  # enumerate 사용
        queue.append((i, p)) # (index, priorities) => [(0, 2), (1, 1), (2, 3), (3, 2)]
        
    while True:
        cur = queue.pop(0) # 가장 앞 pop. (current)
        if any(cur[1] < q[1] for q in queue): # 현재 요소의 우선순위보다 queue에 있는 어떤 요소의 우선순위가 더 큰 게 있다면
            queue.append(cur) # 현재 요소를 가장 뒤에 append
        else: # 현재 요소의 우선순위가 가장 크다면 pop으로 인해 개수 하나가 빠지는 것이므로
            answer += 1 # answer의 index 늘리고
            if cur[0] == location: # 만약 현재 index가 location과 같다면 답
                return answer