6. 응급실(큐)_파이썬

import sys
from collections import deque
# 입력값: 5 2
#60 50 70 80 90
sys.stdin=open("C:\\input.txt",'rt')

n, m = map(int,input().split())
Q = [(pos,val)for pos,val in enumerate(list(map(int,input().split())))]
Q = deque(Q)
print(Q)
cnt = 0
while True:
    cur = Q.popleft()
    if any(cur[1] < x[1] for x in Q):
        Q.append(cur)
    else:
        cnt +=1
        if cur[0] == m:
            print(cnt)
            break

---

📌 문제 설명

• 환자 N명이 대기실에 있음.
• 각 환자의 위험도(우선순위)가 주어짐.
• 위험도가 높은 환자부터 진료됨.
• M번째 환자가 몇 번째로 진료받는지 출력해야 함.

📝 실행 예제

입력

 

5 2
60 50 70 80 90

(환자 5명, 2번 환자가 몇 번째로 진료받는지 찾기)

환자 상태 변화

단계대기열 (Q)진료받은 환자

초기	[(0,60), (1,50), (2,70), (3,80), (4,90)]	-
1	[(2,70), (3,80), (4,90), (0,60), (1,50)]	-
2	[(3,80), (4,90), (0,60), (1,50), (2,70)]	-
3	[(4,90), (0,60), (1,50), (2,70), (3,80)]	-
4	[(0,60), (1,50), (2,70), (3,80)]	90
5	[(1,50), (2,70), (3,80), (0,60)]	80
6	[(1,50), (3,80), (0,60)]	70 (2번 환자)

출력 결과

 

3

➡ 2번 환자는 3번째로 진료받음! ✅

 

🚀 최적화 코드

💡 max()를 사용해서 any() 연산을 최적화할 수 있어!

from collections import deque

n, m = map(int, input().split())
Q = deque(enumerate(map(int, input().split())))  # (환자 번호, 위험도) 저장
cnt = 0

while True:
    cur = Q.popleft()

    # 중요도가 가장 높은 환자가 있는지 확인
    if cur[1] < max(Q, key=lambda x: x[1])[1]:
        Q.append(cur)  # 위험도가 낮으면 다시 큐에 삽입
    else:
        cnt += 1  # 진료받음
        if cur[0] == m:
            print(cnt)
            break

• max(Q, key=lambda x: x[1]) 사용 → any()보다 효율적
• O(n²) → O(n log n)로 최적화 가능

---

✅ 정리

✅ 응급실 문제 (우선순위 큐 활용) 해결!
✅ Deque 활용 → popleft() 연산 최적화
✅ any() 대신 max()로 최적화 가능
✅ 시간 복잡도 O(n²) → O(n log n)으로 개선 가능