[파이썬/알고리즘] 응급실

https://www.inflearn.com/course/%ED%8C%8C%EC%9D%B4%EC%8D%AC-%EC%95%8C%EA%B3%A0%EB%A6%AC%EC%A6%98-%EB%AC%B8%EC%A0%9C%ED%92%80%EC%9D%B4-%EC%BD%94%EB%94%A9%ED%85%8C%EC%8A%A4%ED%8A%B8/dashboard

파이썬 알고리즘 문제풀이 입문(코딩테스트 대비) - 인프런 | 강의

 

해당 강좌에 포함된 문제다.

 

문제

문제를 읽어보면 큐에 값만 넣을게 아니라, 값과 index값을 저장해야 한다.

 

그래서 나는 그냥 2차원 리스트를 deque객체에 넣어줬다.

 

정답

from collections import deque

n, m = map(int, input().split())

arr = list(map(int, input().split()))
que = deque()
for i in range(n) :
    que.append([arr[i], i])

count = 0
while True :
    if que[0] == max(que, key = lambda x:x[0]) :
        count += 1
        poplist = que.popleft()
        if poplist[1] == m :
            print(count)
            break
    else :
        que.append(que.popleft())

---

오늘의 배운 점

 

max, min함수를 쓸 때도 key함수를 쓸 수 있다!