[프로그래머스] 이중우선순위큐 c++

코드 짜기 전에 어떻게 하는 게 제일 효율적일까 고민을 했지만 결국은 그냥 노가다(?)로 하면 됐던 문제.

근데 풀고 나서 다른 사람들 풀이를 보니 완전이진트리의 형태로 자동으로 정렬해주는 multiset 이라는 class가 있었다. 

 

문제설명

이중 우선순위 큐는 다음 연산을 할 수 있는 자료구조를 말합니다.

명령어수신 탑(높이)

I 숫자	큐에 주어진 숫자를 삽입합니다.
D 1	큐에서 최댓값을 삭제합니다.
D -1	큐에서 최솟값을 삭제합니다.

이중 우선순위 큐가 할 연산 operations가 매개변수로 주어질 때, 모든 연산을 처리한 후 큐가 비어있으면 [0,0] 비어있지 않으면 [최댓값, 최솟값]을 return 하도록 solution 함수를 구현해주세요.

 

 

제한사항

• operations는 길이가 1 이상 1,000,000 이하인 문자열 배열입니다.
• operations의 원소는 큐가 수행할 연산을 나타냅니다.
  • 원소는 “명령어 데이터” 형식으로 주어집니다.- 최댓값/최솟값을 삭제하는 연산에서 최댓값/최솟값이 둘 이상인 경우, 하나만 삭제합니다.
• 빈 큐에 데이터를 삭제하라는 연산이 주어질 경우, 해당 연산은 무시합니다.

 

풀이

#include <string>
#include <vector>
#include <algorithm>

using namespace std;

vector<int> array_;

// 숫자 삽입 함수
void array_push(int data) {
    int tmp;
    array_.push_back(data);
    if (array_.size() != 1){
        // array가 비어있지 않은 경우에만 처리
        for (int i=array_.size()-1 ; i>0; i--){
            if (array_[i-1] <= array_[i]) break;
            else {
                // 자기 자리를 찾을 때까지 앞으로 이동
                tmp = array_[i];
                array_[i] = array_[i-1];
                array_[i-1] = tmp;
            }
        }
    }
}
// 최댓값 제거 함수
void pop_max(){
    if (array_.size() != 0) array_.pop_back();
}

// 최솟값 제거 함수
void pop_min(){
    if (array_.size() != 0) array_.erase(array_.begin(), array_.begin() + 1);
}

vector<int> solution(vector<string> operations) {
    vector<int> answer;
    //reverse 함
    reverse(operations.begin(), operations.end());
    while(operations.size() != 0){
        string operation = operations.back();
        operations.pop_back();
        if (operation[0]=='I') {
            int num = stoi(operation.substr(2, operation.length()));
            array_push(num);
        }
        else if (operation[2]=='1') pop_max();
        else pop_min();                         
    }
    
    if (array_.size() == 0){
        // 최종 array가 빈 경우 0,0 넣기
        answer.push_back(0);
        answer.push_back(0);
    }
    else {
        answer.push_back(array_.back());
        answer.push_back(array_.front());
    }
    return answer;
}