[스택/큐] Programmers lv.2 프로세스(python)

문제 설명

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

1. 실행 대기 큐(Queue)에서 대기 중인 프로세스 하나를 꺼냅니다.
2. 큐에 대기 중인 프로세스 중 우선순위가 더 높은 프로세스가 있다면 방금 꺼낸 프로세스를 다시 큐에 넣습니다.
3. 만약 그런 프로세스가 없다면 방금 꺼낸 프로세스를 실행합니다.
   3.1 한 번 실행한 프로세스는 다시 큐에 넣지 않고 그대로 종료됩니다.

예를 들어 프로세스 4개 [A, B, C, D]가 순서대로 실행 대기 큐에 들어있고, 우선순위가 [2, 1, 3, 2]라면 [C, D, A, B] 순으로 실행하게 됩니다.

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

제한사항

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

입출력 예

from collections import deque
def solution(priorities, location):
    queue = []
    answer = 0
    # 왼쪽부터 실행되는데 만약 더 우선순위 높은 프로세스가 있다면 다시 큐에 넣는다.
    for i,prior in enumerate(priorities):
        # 튜플 요소(우선순위)랑 인덱스 같이 저장한다.
        queue.append((prior,i))
    dq = deque(queue)
    while len(dq) > 0: 
        # 만약 우선순위 높은 애가 나간다면 그다음 높은애로 최댓값을 갱신해줘야 함
        maxPriority = max(dq, key=lambda x: x[0])
        # 실행된 애가 maxPriority면 실행해버리고 아니면 큐의 맨 끝으로 다시 넣는다.
        # 어차피 0인덱스로 해도 큐니까 계속 첫번째애가 바뀜
        # 그리고 지금 실행되는 요소가 location 인덱스의 요소 ,그러니까 구해야 하는 요소와 같은지 확인
        if dq[0][0] == maxPriority[0]: # 실행될 애가 우선순위가 가장 높은 애라면
            if dq[0][1] == location:
                answer = answer + 1
                return answer
            else:
                answer = answer + 1
                dq.popleft()
        
        # 실행될 요소가 최댓값이 아니라면
        else:
            dq.append(dq.popleft()) # 빼자마자 넣는다. 

-> 입출력 예를 보면 2번쨰 예시 priorities가 [1,1,9,1,1,1] 일 때 location이 0인 요소 즉 첫번째 요소인 1이 꺼내지고 실행이 안되었을 때 다시 큐로 삽입이 되는데 어떻게 지금 들어간 1이 location 인덱스에 1이라는 걸 증명하는 가에 대해 많은 고민을 했다. 그러다가 flag가 있으면 좋겠다고 생각하고 결국 객체, 즉 튜플자료형을 사용하기로 하였고, 인덱스를 기억해놓으면 되겠다라는 점에 포인트를 두고 문제를 풀었다.