10845: 큐

• toc {:toc}

문제

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

명령은 총 여섯 가지이다.

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

입력

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

출력

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

출처:https://www.acmicpc.net/problem/10845

풀이

큐 구현

STL Queue 사용

#include <bits/stdc++.h>
 
using namespace std;
 
int main(){
    ios::sync_with_stdio(0);
    cin.tie();
    queue<int> q;
    int n;
    cin >> n;
    while(n--){
        string str;
        cin >> str;
        if(str == "push"){
            int p;
            cin >> p;
            q.push(p);
        }
        else if(str == "front"){
            if(q.empty()) cout << -1 << endl;
            else cout << q.front() << endl;
        }
        else if(str == "back"){
            if(q.empty()) cout << -1 << endl;
            else cout << q.back() << endl;
        }
        else if(str == "size") cout << q.size() << endl;
        else if(str == "empty") cout << int(q.empty()) << endl;
        else if(str == "pop"){
            if(q.empty()) cout << -1 << endl;
            else {
                cout << q.front() << endl;
                q.pop();
            }
        }
    }
    return 0;
}

배열로 구현

#include <bits/stdc++.h>
 
using namespace std;
 
const int MX = 1000005;
int dat[MX];
int head=0, tail=0;
 
void push(int x){
    dat[tail++]=x;
}
 
void pop(){
    if(head < tail) head++;
}
 
void size(){
    cout << tail - head << endl;
}
 
int empty(){
    if(head == tail) return 1;
    else return 0;
}
 
void front(){
    if(empty()) cout << -1 << endl;
    else cout << dat[head] << endl;
}
 
void back(){
    if(empty()) cout << -1 << endl;
    else cout << dat[tail-1] << endl;
}
 
int main(){
    ios::sync_with_stdio(0);
    cin.tie();
    int n;
    cin >> n;
    while(n--){
        string str;
        cin >> str;
        if(str == "push"){
            int p;
            cin >> p;
            push(p);
        }
        else if(str == "pop"){
            if(!empty()){
                front();
                pop();
            }
            else cout << -1 << endl;
        }
        else if(str == "size") size();
        else if(str == "empty") cout << empty() << endl;
        else if(str == "front"){
            if(empty()) cout << -1 << endl;
            else cout << dat[head] << endl;
        }
        else if(str == "back"){
            if(empty()) cout << -1 << endl;
            else cout << dat[tail-1] << endl;
        } 
    }
    return 0;
}