백준 1966. 프린터 큐

문제

여러분도 알다시피 여러분의 프린터 기기는 여러분이 인쇄하고자 하는 문서를 인쇄 명령을 받은 ‘순서대로’, 즉 먼저 요청된 것을 먼저 인쇄한다. 여러 개의 문서가 쌓인다면 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 이하의 정수이고, 중요도가 같은 문서가 여러 개 있을 수도 있다.

출력

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

예제 입력 

3
1 0
5
4 2
1 2 3 4
6 0
1 1 9 1 1 1

예제 출력

1
2
5

 

 

---

문제 접근

1.선입선출 (FIFO) 생각하며 queue 사용해야함! 근데 나는 큐가 낯설어서 

stack으로 append()와 pop()을 이용해서 문제를 우선 풀어봤다.

 

2.최대중요도가 될때까지 pop(0)하고 아니면 pop(0)한 값을 append()로 다시 뒤로 보낸다.

while문으로 pop(0)값이 m(내가 뽑고자 하는 번호)값이 될때까지 반복한다.

 

배운 점

• deque(Double-Ended Queue)

리스트에서의 pop(0) 연산은 O(n) 시간 복잡도, 성능이 떨어지지만
deque를 사용한 코드는 collections.deque를 사용하여 큐 기능을 구현하고,

deque는 양쪽 끝에서의 삽입과 삭제 연산이 O(1) 시간 복잡도로 매우 효율적입니다.

 

• enumerate()

enumerate() 함수는 리스트 my_list의 각 요소와 인덱스를 (인덱스, 요소) 형태의 튜플로 반환!

인덱스를 직접 추적해야할 때 매우 유용하다.

 

my_list = ['apple', 'banana', 'cherry']

for idx, fruit in enumerate(my_list):
    print(f"Index {idx}: {fruit}")

---

코드1) stack 사용

 

T = int(input())
for _ in range(T):
    cnt = 0
    n, m = map(int, input().split())  # n: 문서의 개수, m: 몇 번째에 있는지
    doc = list(map(int, input().split()))
    lst = list(range(n))

    while True:
        max_doc = max(doc)  # 가장 높은 중요도를 반복문 내에서 갱신
        if doc[0] < max_doc:
            doc.append(doc.pop(0))
            lst.append(lst.pop(0))
        else:
            doc.pop(0)
            cnt += 1
            if lst.pop(0) == m:  # 현재 출력된 문서가 찾고자 하는 문서인 경우
                print(cnt)
                break

 

코드2) deque 사용

 

from collections import deque

T = int(input())
for _ in range(T):
    cnt = 0
    n, m = map(int, input().split())  # n 문서의 개수, m 몇 번째에 있는지
    doc = list(map(int, input().split()))

    # 문서의 인덱스와 중요도를 함께 큐에 저장
    queue = deque([(i, d) for i, d in enumerate(doc)])
    
    while queue:
        # 현재 큐의 첫 번째 문서를 꺼내서 확인
        cur = queue.popleft()

        # 현재 문서보다 중요도가 높은 문서가 큐에 존재하는지 확인
        if any(cur[1] < other[1] for other in queue):
            queue.append(cur)  # 중요도가 높은 문서가 있으면 현재 문서를 다시 큐에 삽입
        else:
            cnt += 1  # 중요도가 가장 높은 문서를 출력
            if cur[0] == m:  # 출력한 문서가 우리가 찾고자 하는 문서인 경우
                print(cnt)
                break