스택으로 큐 구현하기(c++)

오늘은 스택으로 큐를 구현해보려고 합니다.

총 2개의 스택을 사용하여 큐를 구현할 수 있습니다.

push, pop, isEmpty함수를 구현했습니다.

 

저는 이해를 위해 다음 블로그를 참조하였고 따로 c++로 코드를 작성해보았습니다.

https://tdm1223.tistory.com/44

스택으로 큐 구현하기 / 큐로 스택 구현하기

 

코드 첨부

 

// 스택 2개를 사용해서 큐를 구현하시오. (push, pop, empty 함수 구현)
#include <iostream>
#include <stack>
using namespace std;
 
int n, num;
string s;
stack<int> s1;
stack<int> s2;
 
void push(int data){
    s1.push(data);
}
 
bool isEmpty(){
    if(s1.empty()){
        return true;
    }
 
    return false;
}
 
int pop(){
    if(isEmpty()){
        cout<<"stack underflow"<<"\n";
        return 0;
    }
 
    while(!s1.empty()){
        s2.push(s1.top());
        s1.pop();
    }
 
    int num = s2.top();
    s2.pop();
 
    while(!s2.empty()){
        s1.push(s2.top());
        s2.pop();
    }
 
    return num;
}
 
int main()
{
    cin>>n;
    for(int i=0; i<n; i++){
        cin>>s;
        if(s=="pop"){
            cout<<pop()<<"\n";
        }else if(s=="push"){
            cin>>num;
            push(num);
        }else if(s=="empty"){
            cout<<isEmpty()<<"\n";
        }
    }
 
    return 0;
}