스택&큐(유형정리) - 백준 1966

1966번: 프린터 큐 (acmicpc.net)

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 이하의 정수이고, 중요도가 같은 문서가 여러 개 있을 수도 있다.

출력

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

예제 입력 1 복사

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

예제 출력 1 복사

1
2
5

 

우선순위를 찾아내는 문제이다.

 

문제 자체는 단순하다. 우선순위를 조사해서 찾으려는 인덱스 위치의 값보다 우선순위가 높은 값이 뒤에 있으면 맨 뒤로 보내고 아니면 출력하면 된다.

 

우선순위가 높은 값이 뒤에 있는 경우를 리스트 인덱스를 활용해서 조사하는 방식을 처음에 택하다가 단순히 우선순위가 가장 높은 경우를 만날 때를 뽑아내고 그다음 나머지 경우를 조정해주면 간단히 해결된다는 사실을 깨닫고 알고리즘을 수정했다.

 

※ 알고리즘

1. 테스트케이스 입력받기

2. 케이스 갯수 만큼 for문 내에서 N, M 입력받기

3. 중요도 리스트 입력받기

4. 인덱스 M이 -1이 될 경우(빠져나간 경우)가 아니라면 반복문을 계속해서 시행

5. 중요도 리스트 맨 앞이 중요도가 제일 높은 경우 pop시키고 M을 줄여주고 count를 높여준다.

6. 중요도 높은게 뒤에 있는 else의 경우 중요도를 맨 뒤로 보내준다.

7. 0번째 숫자를 찾는 경우 M을 맨 뒤 바로 앞으로, 0번째 숫자를 찾는 경우가 아니라면 M을 1칸 줄여서 반복을 시행한다.

8. 우선순위 가장 높은 것을 만난 경우 count가 증가할 것이므로 count 변수를 출력시킨다.

 

 

case = int(input())

for i in range(case):
    N, M = map(int,input().split())
    rate = list(map(int, input().split()))
    count = 0

    while M != -1:
        if rate[0] == max(rate):    # 우선순위가 가장 높은 것을 만난 경우
            rate.pop(0)
            M -= 1
            count += 1
        else:                       # 우선순위 높은게 뒤에 있는 경우
            rate.append(rate.pop(0))

            if M == 0:              # 0번째 숫자인 경우
                M = len(rate) - 1   # 맨뒤로
            else:                   # 0번째 숫자 아닌 경우
                M -= 1              # M 1칸 줄이기

    print(count)

 

솔직히 deque를 쓰지 않아 시간초과가 날 수도 있을 것이라고 생각했지만, 기본 알고리즘이라 그런지 깔끔히 통과가 되었다... 휴.......