반응형
문제
정수를 저장하는 큐를 구현한 다음, 입력으로 주어지는 명령을 처리하는 프로그램을 작성하시오.
명령은 총 여섯 가지이다.
- 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 |