백준 1966 [프린터 큐]

출처 : www.acmicpc.net/problem/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

 

 

	#include<iostream>
	#include<queue>
	using namespace std;
	int main() {
	std::ios::sync_with_stdio(false);
	int T, N, M, x, index;
	cin >> T;
	for (int i = 0; i < T; i++) {
	queue<pair<int, int>> q;
	priority_queue<int> pq;
	int count = 0;
	cin >> N >> M;
	for (int j = 0; j < N; j++) {
	cin >> x;
	q.push({ x, j });
	pq.push(x);
	}
	while (1) {
	x = q.front().first;
	index = q.front().second;
	q.pop();
	if (pq.top() == x) {
	count++;
	pq.pop();
	if (index == M)
	{
	cout << count << '\n';
	break;
	}
	}
	q.push({ x, index });
	}
	}
	return 0;
	}

view raw 1966.cpp hosted with ❤ by GitHub

 

 

queue와 priority queue를 이용하여 푸는 문제.

나는 vector와 queue를 이용하여 풀려고 했는데 두 예제에 관하여 제대로 된 답도 안나오고 

정렬을 어떻게 해야할지 모르겠어서 계속 고민하다가 결국 인터넷을 참고하였다.

이번 기회를 통해 priority queue 사용하는 법과 이것을 사용하면 중요도가 높은 순으로 자동 정렬해준다는 사실을 알았다.

 

queue로 먼저 입력을 받고 priority queue에는 중요도만 입력을 해준다.

그런 다음 priority에 입력된 중요도와 queue의 front에 있는 중요도와 비교를 해주며

다를 경우 queue front를 pop하여 다시 맨 뒤로 넣어준다.

같을 경우에는 그 중요도에 해당하는 것이 출력되는 것이기 때문에

count를 증가시키며 queue와 priority queue에서 다 pop해준다.

그 다음, index까지 일치하면 count를 출력 후 해당 테스트케이스를 완료해주면 된다.

나는 계속 정렬에 초점을 맞췄었는데

정렬하지 않아도 priority queue를 이용하여 중요도만 비교해주면 풀 수 있는 문제였다.