[알고리즘]응급실

김도연·2024년 1월 19일

알고리즘

목록 보기
34/56

문제

메디컬 병원 응급실에는 의사가 한 명밖에 없습니다.
응급실은 환자가 도착한 순서대로 진료를 합니다. 하지만 위험도가 높은 환자는 빨리 응급조 치를 의사가 해야 합니다. 이런 문제를 보완하기 위해 응급실은 다음과 같은 방법으로 환자의 진료순서를 정합니다.
• 환자가 접수한 순서대로의 목록에서 제일 앞에 있는 환자목록을 꺼냅니다.
• 나머지 대기 목록에서 꺼낸 환자 보다 위험도가 높은 환자가 존재하면 대기목록 제일 뒤로
다시 넣습니다. 그렇지 않으면 진료를 받습니다.
현재 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

[내 코드]

n,m=map(int,input().split())
a=list(map(int,input().split()))
num=[]
ord=[]
for i in range(n):
    num.append(i)
while a:
    if a[0]<max(a):
        a.append(a.pop(0))
        num.append(num.pop(0))
    else:
        a.pop(0)
        ord.append(num.pop(0))
for i in range(len(ord)):
    if m==ord[i]:
        print(i+1)
        break
  1. 자료구조 큐를 이용하면 쉽게 풀리는 문제.
  2. 입력값을 리스트 형태로 받는다.
  3. a[0]을 통해 입력받은 리스트a의 첫 번째 값과 입력받은 리스트a 중 가장 큰 값과 비교
    1. a[0]<max(a)인 경우는 위험도가 리스트내의 가장 높은 위험도보다 낮기 때문에 문제의 조건2에 따라서 맨 뒤로 이동=>a.append
  4. a[0]>=max(a)인 경우는 위험도가 가장 큰 경우이므로 우선 진료를 받는다.따라서, pop(0)을 통해 리스트에서 제외시킨다.

[해설코드]

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:
	cur=Q.popleft()
    if any(cur[1]<x[1] for x in Q):
    	Q.append(cur)
    else:
    	cnt+=1
        if cur[0]==m:
        	break
print(cnt)

0개의 댓글