0217. [BOJ#1966] 프린터 큐

2 초

128 MB

75210

43299

34113

58.510%

문제

여러분도 알다시피 여러분의 프린터 기기는 여러분이 인쇄하고자 하는 문서를 인쇄 명령을 받은 ‘순서대로’, 즉 먼저 요청된 것을 먼저 인쇄한다. 여러 개의 문서가 쌓인다면 Queue 자료구조에 쌓여서 FIFO - First In First Out - 에 따라 인쇄가 되게 된다. 하지만 상근이는 새로운 프린터기 내부 소프트웨어를 개발하였는데, 이 프린터기는 다음과 같은 조건에 따라 인쇄를 하게 된다.

1. 현재 Queue의 가장 앞에 있는 문서의 ‘중요도’를 확인한다.
2. 나머지 문서들 중 현재 문서보다 중요도가 높은 문서가 하나라도 있다면, 이 문서를 인쇄하지 않고 Queue의 가장 뒤에 재배치 한다. 그렇지 않다면 바로 인쇄를 한다.

예를 들어 Queue에 4개의 문서(A B C D)가 있고, 중요도가 2 1 4 3 라면 C를 인쇄하고, 다음으로 D를 인쇄하고 A, B를 인쇄하게 된다.

여러분이 할 일은, 현재 Queue에 있는 문서의 수와 중요도가 주어졌을 때, 어떤 한 문서가 몇 번째로 인쇄되는지 알아내는 것이다. 예를 들어 위의 예에서 C문서는 1번째로, A문서는 3번째로 인쇄되게 된다.

입력

첫 줄에 테스트케이스의 수가 주어진다. 각 테스트케이스는 두 줄로 이루어져 있다.

테스트케이스의 첫 번째 줄에는 문서의 개수 N(1 ≤ N ≤ 100)과, 몇 번째로 인쇄되었는지 궁금한 문서가 현재 Queue에서 몇 번째에 놓여 있는지를 나타내는 정수 M(0 ≤ M < N)이 주어진다. 이때 맨 왼쪽은 0번째라고 하자. 두 번째 줄에는 N개 문서의 중요도가 차례대로 주어진다. 중요도는 1 이상 9 이하의 정수이고, 중요도가 같은 문서가 여러 개 있을 수도 있다.

출력

각 테스트 케이스에 대해 문서가 몇 번째로 인쇄되는지 출력한다.

---

제출 코드 1

from collections import deque

T = int(input())
for tc in range(1, T+1):
    N, M = map(int, input().split()) # 문서의 개수, 궁금문서 idx
    lst = list(map(int, input().split()))
    queue = deque()

    for idx, i in enumerate(lst):
        queue.append((idx, i))
    
    cnt = 0
    while True:
        if queue[0][1] == max(queue, key=lambda x: x[1])[1]: # 만약 맨앞 데이터가 제일 중요하다면
            cnt += 1 # 뽑기
            if queue[0][0] == M: # 뽑은게 내가 찾는거라면, queue[idx == M][i]
                print(cnt) # 프린트
                break
            else: # 아니라면
                queue.popleft() # 빼기
        else:
            queue.append(queue.popleft()) # 빼서 뒤로넣기

최근 스택을 배워서 써보고 싶었는데 백준을 풀 때는 deque를 사용하고 싶었다.

이 문제에는 자리와 중요도가 같이 있어서 튜플을 사용하고자 했는데 튜플형태로 리스트에 넣는방법을 몰라 검색을 했고,

enumerate라는 내장함수를 알게되었다. 리스트의 요소들에 idx 값을 부여해줘서 매우 편했다.

for 문으로 받아서 튜플로 다시 queue에 넣어주었고 max함수를 통해 제일 중요한 문서를 찾고자 했다.

제출코드 2의 방법이 원래 풀던 방법이지만, max로 풀고싶어서 max내장함수의 사용법과 lambda 사용법을 알아보았다.

새로 배운 개념을 완벽히 쓰는데는 역시 자료구조와 언의 이해가 필요하다고 느꼈다

제출 코드 2

from collections import deque

T = int(input())
for tc in range(1, T+1):
    N, M = map(int, input().split()) # 문서의 개수, 궁금문서 idx
    lst = list(map(int, input().split()))
    queue = deque()

    for idx, i in enumerate(lst):
        queue.append((idx, i))
    
    cnt = 0
    while True:
        impo = -1
        for j in queue: # 최고중요도 탐색
            if j[1] > impo:
                impo = j[1]
        if queue[0][1] == impo: # 만약 맨앞 데이터가 제일 중요하다면
            cnt += 1 # 뽑기
            if queue[0][0] == M: # 뽑은게 내가 찾는거라면, queue[idx == M][i]
                print(cnt) # 프린트
                break
            else: # 아니라면
                queue.popleft() # 빼기
        else:
            queue.append(queue.popleft()) # 빼서 뒤로넣기