본문 바로가기
dev/코딩테스트

[BOJ - 10845] 큐

by dev-everyday 2024. 12. 16.
반응형

문제

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

명령은 총 여섯 가지이다.

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

입력

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

 

출력

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

 

풀이

간단한 큐 문제라고 생각할 수 있다.

물론 진짜 큐 자료구조를 사용한다면 굉장히 쉬운 문제이다.

입력 받을 때 getline을 통해서 "push 3"과 같은 문자열의 공백 상관 없이 입력받기만 하면 된다.

이전에 cin >> n으로 입력 받았으므로 버퍼를 비워주기 위해서 cin.ignore을 사용해서 버퍼를 비워주는 것만 유의하자.

 

두번째 코드는 실제로 이 문제의 의미를 생각하면서 풀어보았다.

이 문제는 실제로 큐를 쓸 줄 아는지가 궁금한 것이 아니다.

배열을 선언하고 큐의 동작과 동일하게 작동하도록 코드를 작성하였다.

queue 구현은 시작점과 끝점을 전역 변수로 들고 해당 값을 기준으로 구현하였다.

 

✅ inline 함수의 장점:

  • 속도 향상: 작은 함수에서 빠른 실행을 기대할 수 있어.
  • 코드 가독성: 코드 중복을 줄이면서도 함수의 간결함을 유지할 수 있어.

 

코드

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

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

    int n;
    cin >> n;
    cin.ignore();

    queue<int> q;

    while (n--) {
        string s;
        getline(cin, s);

        if (s.substr(0, 4) == "push") {
            int num = stoi(s.substr(5));
            q.push(num);
        }
        else if (s == "front") {
            cout << (q.empty() ? -1 : q.front()) << "\n";
        }
        else if (s == "back") {
            cout << (q.empty() ? -1 : q.back()) << "\n";
        }
        else if (s == "size") {
            cout << q.size() << "\n";
        }
        else if (s == "empty") {
            cout << q.empty() << "\n";
        }
        else if (s == "pop") {
            if (q.empty()) {
                cout << -1 << "\n";
            }
            else {
                cout << q.front() << "\n";
                q.pop();
            }
        }
    }

    return 0;
}

 

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

const int MAX_SIZE = 12000;
int q[MAX_SIZE];
int front_idx = 0, rear_idx = 0;

inline void push(int x) {
    q[rear_idx++] = x;
}

inline int back() {
    return (rear_idx > front_idx) ? q[rear_idx - 1] : -1;
}

inline int pop() {
    return (rear_idx == front_idx) ? -1 : q[front_idx++];
}

inline int front() {
    return (rear_idx > front_idx) ? q[front_idx] : -1;
}

inline int size() {
    return rear_idx - front_idx;
}

inline int empty() {
    return (rear_idx == front_idx) ? 1 : 0;
}

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

    int T;
    cin >> T;
    while (T--) {
        string cmd;
        cin >> cmd;
        if (cmd == "push") {
            int value;
            cin >> value;
            push(value);
        }
        else if (cmd == "pop") {
            cout << pop() << "\n";
        }
        else if (cmd == "size") {
            cout << size() << "\n";
        }
        else if (cmd == "empty") {
            cout << empty() << "\n";
        }
        else if (cmd == "front") {
            cout << front() << "\n";
        }
        else if (cmd == "back") {
            cout << back() << "\n";
        }
    }
    return 0;
}
반응형

'dev > 코딩테스트' 카테고리의 다른 글

[BOJ - 1260] DFS와 BFS  (0) 2024.12.26
[BOJ - 2178] 미로 탐색  (2) 2024.12.16
[BOJ 9935] 문자열 폭발  (0) 2024.12.16
[BOJ - 2800] 괄호 제거  (2) 2024.12.16
[BOJ - 9663] N-Queen  (0) 2024.12.13