👉 문제 바로가기
\n큐(Queue) : 먼저 집어 넣은 데이터가 먼저 나오는 FIFO(First In First Oout)구조로 데이터를 저장하는 자료 구조우선 큐 클래스에 전체 노드를 순회하고 최대 값을 구하는 메소드를 추가했다. 그리고 문서들의 중요도를 순서대로 큐에 enqueue하고 몇 번째로 인쇄되는지 알고 싶은 문서는 0으로 enqueue하고 그 문서의 중요도는 따로 변수에 저장했다; 0으로 flag 역할을 수행시키기 위해서였다.
\n큐에 문서들의 중요도를 순서대로 다 저장한 후, 문제에서 요구한 rule대로 de_queue했을 때 중요도를 큐에 현존하는 최대 중요도 값과 비교하면서 인쇄 순서를 count해나갔다. 알고싶은 문서는 0으로 flag해놔서 idx가 변화해도 쉽게 알 수 있다; 중요도는 따로 저장해둔 변수 활용하여 비교했다.
\n이번 풀이법은 개인적으로 마음에 들지 않는다. 이런 방법으로도 비효율적으로 풀 수 있구나, 이래서 자료구조를 빠삭하게 공부해야 하는 구나라고 느끼며 이 포스트를 읽었으면 좋겠다.
\n프린터 큐이기 때문에 큐를 갖고 풀 생각만 했다. 제목에서 고정관념을 갖고 사고의 폭을 넓히지 못했던 것 같다.
\n다음 포스트에서는 우선순위 큐와 힙을 정리하고, 좀 더 효율적이고 깔끔한 방법으로 이 문제를 다시 풀어 볼 예정이다.
\n#include <iostream>\n\ntemplate <typename T>\nstruct Node {\n\tT data;\n\tstruct Node *front;\n\tstruct Node *rear;\n};\n\ntemplate <typename T>\nclass Queue {\nprivate:\n\tNode<T> *head;\n\tNode<T> *tail;\n\tint size;\n\t\npublic:\n\tQueue() : head(nullptr), tail(nullptr) , size(0) {}\n\t~Queue() {\n\t\tif (this->head == nullptr)\n\t\t\treturn;\n\t\t\n\t\tNode<T> *cur = this->tail;\n\t\t\n\t\twhile (cur != nullptr) {\n\t\t\tcur = cur->front;\n\t\t\tdelete this->tail;\n\t\t\tthis->tail = cur;\n\t\t}\n\t\t\n\t\tthis->head = this->tail;\n\t}\n\t\n\tint get_size() {\n\t\treturn this->size;\n\t}\n\t\n\tbool is_empty() {\n\t\treturn this->head == nullptr ? true : false;\n\t}\n\t\n\tT get_max() {\n if (is_empty())\n return -1;\n \n\t\tNode<T> *cur = this->tail;\n\t\tT max = this->tail->data;\n\t\t\n\t\twhile (cur != nullptr) {\n\t\t\tif (cur->data > max)\n\t\t\t\tmax = cur->data;\n\t\t\t\n\t\t\tcur = cur->front;\n\t\t}\n\t\t\n\t\treturn max;\n\t}\n\t\n\tvoid en_queue(T value) {\n\t\tNode<T> *node = new Node<T>;\n\t\tnode->data = value;\n\t\tnode->front = nullptr;\n\t\tnode->rear = nullptr;\n\t\t\n\t\tif (is_empty()) // first push.\n\t\t\tthis->head = node;\n\t\telse {\n\t\t\tnode->front = this->tail;\n\t\t\tthis->tail->rear = node;\n\t\t}\n\t\t\n\t\tthis->tail = node;\n\t\t++size;\n\t}\n\t\n\tT de_queue() {\n\t\tif (is_empty()) // empty queue\n\t\t\treturn -1; // T type에 맞는 error code return.\n\t\telse {\n\t\t\tint pop_val = this->head->data;\n\t\t\t\n\t\t\tif (this->head == this->tail) { // queue with 1 element.\n\t\t\t\tdelete this->head;\n\t\t\t\tthis->head = this->tail = nullptr;\n\t\t\t}\n\t\t\telse { // queue with more than 2 elements.\n\t\t\t\tNode<T> *node = new Node<T>;\n\t\t\t\t\n\t\t\t\tnode = this->head->rear;\n\t\t\t\t\n\t\t\t\tdelete this->head;\n\t\t\t\t\n\t\t\t\tthis->head = node;\n\t\t\t\tthis->head->front = nullptr;\n\t\t\t}\n\t\t\t\n\t\t\t--size;\n\t\t\treturn pop_val;\n\t\t}\n\t}\n};\n\nint main() {\n\tstd::ios::sync_with_stdio(false);\n\tstd::cin.tie(NULL); \n\tstd::cout.tie(NULL);\n\t\n\tint n;\n\t\n\tstd::cin >> n;\n\t\n\tfor (int i = 0; i < n; ++i) {\n\t\tint n_doc, m, m_priority, priority, cnt = 0;\n\t\tQueue<int> queue;\n\t\t\n\t\tstd::cin >> n_doc >> m;\n\t\t\n\t\tfor (int j = 0; j < n_doc; ++j) {\n\t\t\tstd::cin >> priority;\n\t\t\t\n\t\t\tif (j == m) {\n\t\t\t\tqueue.en_queue(0);\n\t\t\t\tm_priority = priority;\n\t\t\t}\n\t\t\telse\n\t\t\t\tqueue.en_queue(priority);\n\t\t}\n\t\t\n\t\tif (n_doc == 1) {\n\t\t\tstd::cout << 1 << '\\n';\n\t\t\tcontinue;\n\t\t}\n\t\t\n\t\twhile (m_priority != -1) {\n\t\t\tpriority = queue.de_queue();\n\t\t\t\n\t\t\tif (priority == 0) { // 궁금한 문서\n\t\t\t\tif (m_priority < queue.get_max())\n\t\t\t\t\tqueue.en_queue(priority);\n\t\t\t\telse {\n\t\t\t\t\t++cnt;\n\t\t\t\t\tm_priority = -1;\n\t\t\t\t}\n\t\t\t}\n\t\t\telse { // 궁금하지 않은 나머지 일반 문서들\n\t\t\t\tif (priority < queue.get_max() || priority < m_priority) {\n\t\t\t\t\tqueue.en_queue(priority);\n\t\t\t\t\tcontinue;\n\t\t\t\t}\n\t\t\t\t\n\t\t\t\t++cnt;\n\t\t\t}\n\t\t}\n\t\t\n\t\tstd::cout << cnt << '\\n';\n\t}\n\t\n\treturn 0;\n}