18258번 - 큐2

문제

정수를 저장하는 큐를 구현한 다음, 입력으로 주어지는 명령을 처리하는 프로그램을 작성하시오.

명령은 총 여섯 가지이다.

• push X: 정수 X를 큐에 넣는 연산이다.
• pop: 큐에서 가장 앞에 있는 정수를 빼고, 그 수를 출력한다. 만약 큐에 들어있는 정수가 없는 경우에는 -1을 출력한다.
• size: 큐에 들어있는 정수의 개수를 출력한다.
• empty: 큐가 비어있으면 1, 아니면 0을 출력한다.
• front: 큐의 가장 앞에 있는 정수를 출력한다. 만약 큐에 들어있는 정수가 없는 경우에는 -1을 출력한다.
• back: 큐의 가장 뒤에 있는 정수를 출력한다. 만약 큐에 들어있는 정수가 없는 경우에는 -1을 출력한다.

입력

첫째 줄에 주어지는 명령의 수 N (1 ≤ N ≤ 2,000,000)이 주어진다. 둘째 줄부터 N개의 줄에는 명령이 하나씩 주어진다. 주어지는 정수는 1보다 크거나 같고, 100,000보다 작거나 같다. 문제에 나와있지 않은 명령이 주어지는 경우는 없다.

출력

출력해야하는 명령이 주어질 때마다, 한 줄에 하나씩 출력한다.

'

2022.04.25 - [C++] - cout / cin 최적화

 

#include <iostream>
#define QUEUE_SIZE 2000000
using namespace std;

// push X
// pop
// size
// empty
// front
// back

class Queue
{
public:
    Queue();
    ~Queue();

    void Push(int x);
    int Pop();
    int Size();
    bool IsEmpty();
    int GetFront();
    int GetBack();

private:
    int front, back;
    int *array;
};

int main(void)
{    
    ios::sync_with_stdio(false);
    cin.tie(NULL);

    int repeat;
    cin>>repeat;

    Queue q;
    Queue result_q;
    while(repeat--)
    {        
        string order;
        cin>>order;

        if(order == "push")
        {
            int x;
            cin>>x;
            q.Push(x);
        }
        else if(order == "pop")
        {
            result_q.Push(q.Pop());
        }
        else if(order == "size")
        {
            result_q.Push(q.Size());
        }
        else if(order == "empty")
        {
            result_q.Push(q.IsEmpty());
        }
        else if(order == "front")
        {   
            result_q.Push(q.GetFront());
        }
        else if(order == "back")
        {
            result_q.Push(q.GetBack());
        }
    }

    while(!result_q.IsEmpty())
    {
        cout<<result_q.Pop()<<"\n";
    }
    return 0;
}

Queue::Queue() : front(0), back(-1)
{
    array = new int[QUEUE_SIZE];
}
Queue::~Queue()
{ 
    delete[] array;
}

void Queue::Push(int x)
{
    array[++back] = x;
}
int Queue::Pop()
{
    if(IsEmpty())
        return -1;
    
    return array[front++];
}

int Queue::Size()
{
    int size = back-front+1;
    return size;
}

bool Queue::IsEmpty()
{  
    return (front>back);
}
int Queue::GetFront()
{
    if(IsEmpty())
        return -1;
    return array[front];
}
int Queue::GetBack()
{
    if(IsEmpty())
        return -1;
    return array[back];
}