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

풀이

내림차순으로 정렬되어 최댓값부터 나오는 q

오름차순으로 정렬되어 최솟값부터 나오는 q2

최댓값을 삭제할땐 q에서 pop해주고, 최솟값을 삭제할때 q2에서 pop해준다.

들어온 횟수와 삭제된 횟수가 같다면 q와 q2를 모두 비워준다. (이미 모든 원소가 삭제된 것이므로)

#include <string>
#include <vector>
#include <queue>
#include <iostream>
using namespace std;

vector<int> solution(vector<string> operations) {
    vector<int> answer;
    int cnt = 0;
    priority_queue<int>q;
    priority_queue<int,vector<int>,greater<int>>q2;
    int re_cnt=0;
    for(auto op : operations){
        if(op[0]=='I'){
            cnt++;
            q.push(stoi(op.substr(2)));
            q2.push(stoi(op.substr(2)));
           
        }
        
        
        //최솟값 삭제
       if(!q2.empty() && op[0]=='D' && op[2]=='-'){
           q2.pop();
           re_cnt++;
        }
        
        //최댓값 삭제
        if(!q.empty() && op[0]=='D' && op[2]!='-'){
            q.pop();
            re_cnt++;
        }
        
        if(re_cnt==cnt){
            while(!q.empty())
                q.pop();
            while(!q2.empty())
                q2.pop();
        }
    }
    if(cnt-re_cnt==0){
         answer.push_back(0);
         answer.push_back(0);
    }
    else{
        answer.push_back(q.top());
        answer.push_back(q2.top());
    }

        
    return answer;
}

 

다른 사람의 풀이

원소가 들어올때 cnt를 증가하고 삭제할땐 감소해서 cnt 로만 원소가 남았는지 확인했다.

 

stringstream ss(operations[i]);
 ss >> char형 변수
 ss >> number형 변수

이렇게 입력을 받아서 ss에서 char와 number를 할당받았다.