BOJ - 28278번 - 스택 2

---

import sys
input = sys.stdin.readline
commands = int(input())


class Stack:
    def __init__(self):
        self.items = []

    def push(self, item):
        self.items.append(item)

    def pop(self):
        if not self.is_empty():
            return self.items.pop()
        return -1

    def peek(self):
        if not self.is_empty():
            return self.items[-1]
        return -1

    def is_empty(self):
        return len(self.items) == 0

    def size(self):
        return len(self.items)


s = Stack()

for _ in range(commands):
    command = input()
    if command[0] == "1":
        num = int(command[2:])
        s.push(num)
    elif command[0] == "2":
        print(s.pop())
    elif command[0] == "3":
        print(s.size())
    elif command[0] == "4":
        print(int(s.is_empty()))
    elif command[0] == "5":
        print(s.peek())

 

정리

가독성을 위해 class를 사용하여 스택을 구현한다.

전체 시간복잡도는 command(명령어) 만큼 실행하므로 O(N)

 

 

 

출처 : https://www.acmicpc.net/problem/28278