[C++] BOJ (백준) 10828 : 스택

문제

10828번: 스택 (acmicpc.net)

10828번: 스택

코드 1 (std::stack)

#include <iostream>
#include <stack>

using namespace std;

stack<int> st;

void push(int num) {
  st.emplace(num);
}

int pop() {
  if (st.empty()) return -1;

  int res = st.top();
  st.pop();
  return res;
}

int size() {
  return (int) st.size();
}

int empty() {
  return st.empty();
}

int top() {
  if (st.empty()) return -1;

  return st.top();
}

int main() {
  ios_base::sync_with_stdio(false);
  cin.tie(nullptr);

  int n;
  cin >> n;
  while (n--) {
    string cmd;
    cin >> cmd;
    if (cmd == "push") {
      int num;
      cin >> num;
      push(num);
    } else if (cmd == "pop") {
      cout << pop() << '\n';
    } else if (cmd == "size") {
      cout << size() << '\n';
    } else if (cmd == "empty") {
      cout << empty() << '\n';
    } else if (cmd == "top") {
      cout << top() << '\n';
    }
  }
  return 0;
}

 

설명

stack 헤더의 std::stack을 이용해 손쉽게 풀 수 있다.

 

코드 2 (직접 구현)

#include <iostream>

using namespace std;

struct LinkedList {
  struct Node {
    int data;
    Node *next;

    explicit Node(int data) : data(data), next(nullptr) {}
  };

  Node *head = nullptr;

  [[nodiscard]] bool isEmpty() const {
    return !head;
  }

  void addFront(int data) {
    Node *pNode = new Node(data);
    if (isEmpty()) head = pNode;
    else {
      pNode->next = head;
      head = pNode;
    }
  }

  void deleteFront() {
    head = head->next;
  }
};

struct Stack : LinkedList {
  void push(int num) {
    addFront(num);
  }

  bool empty() {
    return isEmpty();
  }

  int top() {
    if (empty()) return -1;

    return head->data;
  }

  int pop() {
    if (empty()) return -1;

    int res = top();
    deleteFront();
    return res;
  }

  int size() {
    int cnt = 0;
    Node *pNode = head;
    while (pNode) {
      cnt++;
      pNode = pNode->next;
    }
    return cnt;
  }
};

int main() {
  ios_base::sync_with_stdio(false);
  cin.tie(nullptr);

  Stack st;
  int n;
  cin >> n;
  while (n--) {
    string cmd;
    cin >> cmd;
    if (cmd == "push") {
      int num;
      cin >> num;
      st.push(num);
    } else if (cmd == "pop") {
      cout << st.pop() << '\n';
    } else if (cmd == "size") {
      cout << st.size() << '\n';
    } else if (cmd == "empty") {
      cout << st.empty() << '\n';
    } else if (cmd == "top") {
      cout << st.top() << '\n';
    }
  }
  return 0;
}

 

설명

Linked List를 구현한 후 상속하여 Stack을 구현하였다.