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

문제

---

https://school.programmers.co.kr/learn/courses/30/lessons/42628

풀이

---

이중 우선순위 큐를 구하기 위해, 각 배열의 문자를 나눠 명령어와 숫자 배열을 생성했다. 생성된 배열의 값에 따라 answer에 적용하여 모든 연산을 처리한 후 큐가 비어있으면 (0,0) 비어있지 않으면 (최댓값, 최솟값)을 return한다.

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

using namespace std;

vector<int> solution(vector<string> operations) 
{
    vector<int> answer;

    vector<string> oper;
    vector<string> number;

    for (int i = 0; i < operations.size(); i++)
    {
        stringstream sstream(operations[i]);
        string stringBuffer;
        getline(sstream, stringBuffer, ' ');
        oper.push_back(stringBuffer);

        getline(sstream, stringBuffer, ' ');
        number.push_back(stringBuffer);
    }


    for (int i = 0; i < oper.size(); i++)
    {
        if (oper[i] == "I")
        {
            answer.push_back(stoi(number[i]));
        }

        else
        {
            if (!answer.size() == 0)
            {
                if (number[i] == "1")
                {
                    //최댓값 삭제
                    int max_index = max_element(answer.begin(), answer.end()) - answer.begin();
                    answer.erase(answer.begin() + max_index);
                }

                else
                {
                    //최솟값 삭제
                    int min_index = min_element(answer.begin(), answer.end()) - answer.begin();
                    answer.erase(answer.begin() + min_index);
                }
            }

        }
    }

    if (answer.size() == 0)
    {
        answer.push_back(0);
        answer.push_back(0);
    }

    else
    {
        vector<int> result;

        int max = *max_element(answer.begin(), answer.end());
        result.push_back(max);
        int min = *min_element(answer.begin(), answer.end());
        result.push_back(min);

        answer = result;
    }

    return answer;
}