응급실(큐)

자료구조 활용 (스택, 큐, 해쉬, 힙)

문제 ✏️

응급실(큐)

메디컬 병원 응급실에는 의사가 한 명밖에 없습니다.
응급실은 환자가 도착한 순서대로 진료를 합니다. 하지만 위험도가 높은 환자는 빨리 응급조치를 의사가 해야 합니다. 이런 문제를 보완하기 위해 응급실은 다음과 같은 방법으로 환자의 진료순서를 정합니다.

• 환자가 접수한 순서대로의 목록에서 제일 앞에 있는 환자목록을 꺼냅니다.
• 나머지 대기 목록에서 꺼낸 환자 보다 위험도가 높은 환자가 존재하면 대기목록 제일 뒤로 다시 넣습니다. 그렇지 않으면 진료를 받습니다.
즉 대기목록에 자기 보다 위험도가 높은 환자가 없을 때 자신이 진료를 받는 구조입니다.

현재 N명의 환자가 대기목록에 있습니다.
N명의 대기목록 순서의 환자 위험도가 주어지면, 대기목록상의 M번째 환자는 몇 번째로 진료를 받는지 출력하는 프로그램을 작성하세요.
대기목록상의 M번째는 대기목록의 제일 처음 환자를 0번째로 간주하여 표현한 것입니다.

▣ 입력설명
첫 줄에 자연수 N(5<=N<=100)과 M(0<=M<N) 주어집니다.
두 번째 줄에 접수한 순서대로 환자의 위험도(50<=위험도<=100)가 주어집니다.
위험도는 값이 높을 수록 더 위험하다는 뜻입니다. 같은 값의 위험도가 존재할 수 있습니다.

▣ 출력설명
M번째 환자의 몇 번째로 진료받는지 출력하세요.

▣ 입력예제 1
5 2
60 50 70 80 90

▣ 출력예제 1
3

▣ 입력예제 2
6 0
60 60 90 60 60 60

▣ 출력예제 2
5

코드 💻

import sys
from collections import deque
#sys.stdin=open("input.txt", "rt")  # read text
n, m = map(int, input().split())
Q = [(pos, val) for pos, val in enumerate(list(map(int, input().split())))] # 인덱스, 위험도를 튜플로 받음
Q = deque(Q)
cnt = 0
while True: # n번째 사람이 진료받은 순간 종료
    cur = Q.popleft()
    if any(cur[1] < x[1] for x in Q):   # 하나를 이미 pop함
        Q.append(cur)
    else:
        cnt += 1   # 진료받는 횟수 증가시킴
        if cur[0] == m:
            print(cnt)
            break

참고

• 인프런 : 파이썬 알고리즘 문제 풀이