백준 10866번: 덱
by Jm Park
문제
정수를 저장하는 덱(Deque)를 구현한 다음, 입력으로 주어지는 명령을 처리하는 프로그램을 작성하시오.
명령은 총 여덟 가지이다.
- push_front X: 정수 X를 덱의 앞에 넣는다.
- push_back X: 정수 X를 덱의 뒤에 넣는다.
- pop_front: 덱의 가장 앞에 있는 수를 빼고, 그 수를 출력한다. 만약, 덱에 들어있는 정수가 없는 경우에는 -1을 출력한다.
- pop_back: 덱의 가장 뒤에 있는 수를 빼고, 그 수를 출력한다. 만약, 덱에 들어있는 정수가 없는 경우에는 -1을 출력한다.
- size: 덱에 들어있는 정수의 개수를 출력한다.
- empty: 덱이 비어있으면 1을, 아니면 0을 출력한다.
- front: 덱의 가장 앞에 있는 정수를 출력한다. 만약 덱에 들어있는 정수가 없는 경우에는 -1을 출력한다.
- back: 덱의 가장 뒤에 있는 정수를 출력한다. 만약 덱에 들어있는 정수가 없는 경우에는 -1을 출력한다.
입력
첫째 줄에 주어지는 명령의 수 N (1 ≤ N ≤ 10,000)이 주어진다. 둘쨰 줄부터 N개의 줄에는 명령이 하나씩 주어진다. 주어지는 정수는 1보다 크거나 같고, 100,000보다 작거나 같다. 문제에 나와있지 않은 명령이 주어지는 경우는 없다.
출력
출력해야하는 명령이 주어질 때마다, 한 줄에 하나씩 출력한다.
예제 입력/출력
| 입력 | 출력 |
|---|---|
| 15 push_back 1 push_front 2 front back size empty pop_front pop_back pop_front size empty pop_back push_front 3 empty front |
2 1 2 0 2 1 -1 0 1 -1 0 3 |
| 22 front back pop_front pop_back push_front 1 front pop_back push_back 2 back pop_front push_front 10 push_front 333 front back pop_back pop_back push_back 20 push_back 1234 front back pop_back pop_back |
-1 -1 -1 -1 1 1 2 2 333 10 10 333 20 1234 1234 20 |
코드
#include <iostream>
#include <vector>
#include <string>
using namespace std;
int main() {
int N;
string str;
vector<int> que;
scanf("%d", &N);
getchar(); // 개행문자 버퍼 비워줌
while (N--) {
getline(cin, str);
if (str.find("push_back") != string::npos) {
str.erase(0, 10);
que.push_back(stoi(str));
}
else if (str.find("push_front") != string::npos) {
str.erase(0, 10);
que.insert(que.begin(), stoi(str));
}
else if (str.find("pop_front") != string::npos) {
if (que.empty())
printf("-1\n");
else {
printf("%d\n", que.front());
que.erase(que.begin());
}
}
else if (str.find("pop_back") != string::npos) {
if (que.empty())
printf("-1\n");
else {
printf("%d\n", que.back());
que.pop_back();
}
}
else if (str.find("size") != string::npos) {
que.empty() ? printf("0\n") : printf("%d\n", que.size());
}
else if (str.find("empty") != string::npos) {
que.empty() ? printf("1\n") : printf("0\n");
}
else if (str.find("front") != string::npos) {
que.empty() ? printf("-1\n") : printf("%d\n", que.front());
}
else if (str.find("back") != string::npos) {
que.empty() ? printf("-1\n") : printf("%d\n", que.back());
}
}
return 0;
}
풀이
8가지 조건에 맞춰서 각 조건에 맞게 if-else문을 사용하여 구현하면 쉽다.
이 문제의 경우 C++ STL vector(링크: tistory참고)를 이용하면 굉장히 빠르게 풀린다.
string을 사용하여 해당 문자열이 포함되어있는지 find() 함수를 사용할 경우 if-else문의 순서가 중요하다.
이 때 front와 back을 앞쪽에 써버리면 pop_back/push_back 등 이 앞에서 같이 걸려버리기 때문에 순서만 유의하면 된다.
Subscribe via RSS
