[10828] 스택 (Python)

링크 : https://www.acmicpc.net/problem/10828

10828번: 스택

문제

분류

• 자료구조
• 스택

생각 정리

• 입력을 꽤 많이 받으므로 input()이 아닌 readline()으로 입력을 받아야 겠다.
• stack(리스트)에 쓸 수 있는 push, pop, top, size, empty 문법을 생각해본다. 큐와는 또 다르므로 주의!
  • push : append()
  • pop : pop()
  • top : [-1]  (따로 top()이 아니라 그냥 맨 위의 것을 빼면 된다)
  • size : len()
  • empty : len(stack) == 0 ?  (isEmpty()가 되는줄 알았는데 리스트는 안된다.)

전체 코드는 다음과 같다!

import sys
stack = []

num = int(sys.stdin.readline())  # 숫자 한개 입력시에 split()후에 int() 안됨.
for i in range(num):
    s = list(sys.stdin.readline().split())
    if s[0] == 'push':
        stack.append((int(s[1])))
    elif s[0] == 'pop':
        if stack:
            print(stack.pop())
        else:
            print(-1)
    elif s[0] == 'size':
        print(len(stack))
    elif s[0] == 'empty':
        if len(stack)==0: # list는 stack.isEmpty()가 안됨
            print(1)
        else:
            print(0)
    elif s[0] == 'top':
        if stack:
            print(stack[-1])
        else:
            print(-1)

#stack
# 입력 : append
# 최신 값 빼기 : pop
# 크기 : len
# 최신값 확인 : [-1]