티스토리 뷰

스택


 

시간 제한 메모리 제한 제출 제출 정답 맞힌 사람 정답 비율
0.5 초 (추가 시간 없음) 256 MB 239124 85796 62411 37.271%

문제

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

명령은 총 다섯 가지이다.

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

 

입력

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

출력

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

예제 입력 1 복사

14
push 1
push 2
top
size
empty
pop
pop
pop
size
empty
pop
push 3
empty
top

예제 출력 1 복사

2
2
0
2
1
-1
0
1
-1
0
3

예제 입력 2 복사

7
pop
top
push 123
top
pop
top
pop

예제 출력 2 복사

-1
-1
123
123
-1
-1

 


후기

이번 문제는 구현 자체는 굉장히 쉽기 때문에 문제가 없었지만, 문제를 제대로 읽지 않아 시간이 걸렸다.

 

각 기능은 예외처리 및 결과 값에 따라 다른 출력 값을 내야한다.

이부분에서 놓친 정보가 많아 제대로 출력하지 못했었다.

 

#include <iostream>
#include <sstream>
#include <string>
#include <vector>
#include <stack>

using namespace std;


stack<int> commandStack;

void Printing(int numeric) {
    cout << numeric << '\n';
}

void StackSize() {
    Printing(commandStack.size());
}

void StackTop() {
    Printing(commandStack.empty() ? -1 : commandStack.top());
}

void StackPop() {
    StackTop();
    if (!commandStack.empty()) {
        commandStack.pop();
    }
}

void StackEmpty() {
    Printing(commandStack.empty() ? 1 : 0);
}

int main(void) {
    ios::sync_with_stdio(0);
    cin.tie(0); cout.tie(0);
    
    int loop = 0;
    cin >> loop;
    cin.ignore();

    while (loop--) {
        string command;
        getline(cin, command);
        istringstream iss(command);
        
        string detail = "";
        while (iss >> detail) {
            if (detail == "push") continue;
            
            if (detail == "top") {
                StackTop();
                break; 
            }
            else if (detail == "size") {
                StackSize();
                break; 
            }
            else if (detail == "pop") { 
                StackPop();
                break; 
            }
            else if (detail == "empty") { 
                StackEmpty();
                break;
            }
            else {
                commandStack.push(stoi(detail));
            }
        }
    }
    
    return 0;
}

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

[BOJ/C++] 1463::1로 만들기 (DP)  (0) 2023.11.27
[BOJ/C++] 1026::보물  (1) 2023.11.26
[BOJ/C++] 1027::고층 빌딩  (0) 2023.11.07
[BOJ/C++] 1926::그림  (0) 2023.11.06
[BOJ/C++] 1004::어린왕자  (0) 2023.11.05
공지사항
최근에 올라온 글
최근에 달린 댓글
Total
Today
Yesterday
링크
«   2025/05   »
1 2 3
4 5 6 7 8 9 10
11 12 13 14 15 16 17
18 19 20 21 22 23 24
25 26 27 28 29 30 31
글 보관함