여러분도 알다시피 여러분의 프린터 기기는 여러분이 인쇄하고자 하는 문서를 인쇄 명령을 받은 ‘순서대로’, 즉 먼저 요청된 것을 먼저 인쇄한다. 여러 개의 문서가 쌓인다면 Queue 자료구조에 쌓여서 FIFO - First In First Out - 에 따라 인쇄가 되게 된다. 하지만 상근이는 새로운 프린터기 내부 소프트웨어를 개발하였는데, 이 프린터기는 다음과 같은 조건에 따라 인쇄를 하게 된다.
- 현재 Queue의 가장 앞에 있는 문서의 ‘중요도’를 확인한다.
- 나머지 문서들 중 현재 문서보다 중요도가 높은 문서가 하나라도 있다면, 이 문서를 인쇄하지 않고 Queue의 가장 뒤에 재배치 한다. 그렇지 않다면 바로 인쇄를 한다.
예를 들어 Queue에 4개의 문서(A B C D)가 있고, 중요도가 2 1 4 3 라면 C를 인쇄하고, 다음으로 D를 인쇄하고 A, B를 인쇄하게 된다.
여러분이 할 일은, 현재 Queue에 있는 문서의 수와 중요도가 주어졌을 때, 어떤 한 문서가 몇 번째로 인쇄되는지 알아내는 것이다. 예를 들어 위의 예에서 C문서는 1번째로, A문서는 3번째로 인쇄되게 된다.
입력
첫 줄에 test case의 수가 주어진다. 각 test case에 대해서 문서의 수 N(100이하)와 몇 번째로 인쇄되었는지 궁금한 문서가 현재 Queue의 어떤 위치에 있는지를 알려주는 M(0이상 N미만)이 주어진다. 다음줄에 N개 문서의 중요도가 주어지는데, 중요도는 1 이상 9 이하이다. 중요도가 같은 문서가 여러 개 있을 수도 있다. 위의 예는 N=4, M=0(A문서가 궁금하다면), 중요도는 2 1 4 3이 된다.
결국 우선순위로 뽑아져 나오는 것이므로 우선순위 큐나 벡터로 정렬을 해줄 필요가 있음. 기본 아이디어는 정렬된 우선순위 큐를 기준으로 큐에는 value와 index가 쌍으로 저장되어 front부터 value 값이 우선순위 큐의 top값과 일치하는지 비교 일치 하지 않으면 queue의 뒤에는 자신보다 큰 값이 존재 한다는 뜻이므로 queue를 pop시키고 다시 그 value와 index를 push 시켜서 뒤로 이동 시킨다. 만약 일치 한다면 우선순위 큐와 큐를 각각 pop시킨후 count값을 증가시킴. count는 자신이상의 값이 몇개 존재하는지 세는 변수. 만약 이때 사용자가 찾기 원하는 인덱스가 현재의 인덱스라면 count를 출력시키고 아니라면 우선순위 큐와 큐가 삭제된 시점에서 다시 반복한다. 카운트 값만 1증가한셈.
카운트가 출력되는 것은 단순히 값만 같은 경우가 아니라 인덱스 까지 같은 경우에 한해서 출력
#include <include>
#include<queue>
using namespace std;
int main(void)
{
int N;
cin >> N;
for(int i = 0; i<N; i++)
{
int a, b, c;
int count = 0;
priority_queue pq;
queue<pair<int,int> > que;
cin >> a>> b;
for(int j=0; j<a; j++)
{
int x;
cin >> x;
que.push({x,j});
pq.push(x);
}
while(!pq.empty())
{
int value = que.front().first;
int index = que.front().second;
if(pq.top() == value)
{
pq.pop();
que.pop();
count++;
if(b == index)
{
cout<< count <<endl;
break;
}
}
else{
que.pop();
que.push({value,index});
}
}
}
}
'Algorithm' 카테고리의 다른 글
백준 10822 더하기 (0) | 2019.07.18 |
---|---|
백준 5430 AC (0) | 2019.07.17 |
백준 9012 괄호 (0) | 2019.07.16 |
백준 1753 최단경로 (메모리초과) (0) | 2019.07.14 |
백준 1197번 최소 스패닝 트리 (0) | 2019.07.12 |