[Baekjoon] 10828. 스택(Silver 4)[Python]

10828번: 스택

 

문제

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

 

1. push X : 정수 X를 스택에 넣는 연산이다.

2. pop : 스택에서 가장 위에 있는 정수를 빼고, 그 수를 출력(만약, 스택이 비어있다면 -1을 출력)

3. size : 스택에 들어있는 정수의 개수 출력

4. empty : 스택이 비어있으면 1, 아니면 0을 출력

5. top : 스택에 가장 위에 있는 정수를 출력(만약, 스택이 비어있다면 -1을 출력)

 

풀이

파이썬의 배열은 stack에서 사용하는 pop, peek 기능을 쉽게 구현할 수 있기 때문에

문제를 푸는데 어려움 없습니다.

 

코드

import sys
input = sys.stdin.readline

n = int(input()) # 명령의 수
operator = [input() for _ in range(n)]

stack = []
answer = ''
for op in operator:
  cp = op.split()

  if len(cp) == 2: # push
    num = cp[1]
    stack.append(int(num))
  else: # 나머지 명령
    cp = cp[0]
    if cp == 'pop':
      answer += ("%d\n" % (-1 if len(stack) == 0 else stack.pop(-1)))
    elif cp == 'size':
      answer += ("%d\n" % len(stack))
    elif cp == 'empty':
      answer += ("%d\n" % (1 if len(stack) == 0 else 0))
    else:
      answer += ("%d\n" % (-1 if len(stack) == 0 else stack[-1]))

print(answer)