1 minute read

스택(자료구조)

10828번

문제

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

명령은 총 다섯 가지이다.

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

입력

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

본인 정답 코드

#include <iostream>
using namespace std;

#define MAX 10000

class stack_problem {
private:
    int Array[MAX];
    int top;
public:
    stack_problem();
    void push(int push_num);
    int empty();
    int pop();
    int size();
    int _top();
};

stack_problem::stack_problem() {
    top = 0;
}
void stack_problem::push(int push_num) {
    Array[top] = push_num;
    top++;
}
int stack_problem::empty() {
    if (Array[0] == '\0') {
        return 1;
    }
    else return 0;
}
int stack_problem::pop() {
    if (this->empty() == 1)
        return -1;
    else {
        int tmp = this->Array[this->top - 1];
        this->Array[this->top - 1] = '\0';
        top--;
        return tmp;
    }
}
int stack_problem::size() {
    return this->top;
}
int stack_problem::_top() {
    if (this->empty() == 1)
        return -1;
    else
        return this->Array[top - 1];
}

int main() {

    stack_problem stp;

    int input;
    cin >> input;

    string command;
    for (int i = 0; i < input; i++) {
        cin >> command;
        if (command == "push") {
            int push_num;
            cin.ignore();
            cin >> push_num;

            stp.push(push_num);
        }
        else if (command == "empty") {
            cout << stp.empty() << endl;
        }
        else if (command == "pop") {
            cout << stp.pop() << endl;
        }
        else if (command == "size") {
            cout << stp.size() << endl;
        }
        else if (command == "top") {
            cout << stp._top() << endl;
        }
    }
}

c++에 stack관련 함수들을 사용하면 빠르게 해결할 수 있지만,

클래스를 통해 c++을 복기하며 직접 만들어보았다. 약간의 하드코딩 느낌도 있어 다른 제출자분들 보다 코드량이 길다는 것을 확인했었다.

📣
Baekjoon에서 맞은 코드만 업데이트합니다.
본 포스팅의 풀이 언어 및 개발 환경 : C++, VScode
풀이에 대한 문제나 궁금한 점은 Comments를 작성해주시면, 많은 도움이 됩니다.💡

Leave a comment