baekjoon. 스택 (10828) [python][Silver IV]

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

 

 

 

문제풀이 

import sys

stack = [] # stack = array : 복잡하게 클래스로 만들 필요 없다 
def push(data):
  stack.append(data)

def pop():
  if (empty()):
    print(-1)
  else:
    lastIndex = len(stack) -1
    print(stack.pop(lastIndex)) # remove (value), pop (index)

def size():
  print(len(stack))

def empty():
  if len(stack) == 0:
    return 1
  else:
    return 0

def top():
  if (empty()):
    print(-1)
  else:
    lastIndex = len(stack) -1
    print(stack[lastIndex])

# 명령어 수 받기 
num = int(sys.stdin.readline()) #input 사용할 시 시간 초과 
for i in range(num):
  str = sys.stdin.readline()
  strArray = str.split()

  command = strArray[0]
  if len(strArray) > 1:
    args = strArray[1]

  if (command == 'push'):
    push(args)
  elif (command == 'pop'):
    pop()
  elif (command == 'size'):
    size()
  elif (command == 'empty'):
    print(empty())
  elif (command == 'top'):
    top()

 

---

 

[1] python에서 stack은 list 형태로 구현가능하다. 

ㄴ 공식문서 : https://docs.python.org/ko/3/tutorial/datastructures.html

> list.append(x)

리스트의 끝에 항목을 더합니다. a[len(a):] = [x] 와 동등합니다.

> list.remove(x)

리스트에서 값이 x 와 같은 첫 번째 항목을 삭제합니다. 그런 항목이 없으면 ValueError를 일으킵니다.

> list.pop([i])

리스트에서 주어진 위치에 있는 항목을 삭제하고, 그 항목을 돌려줍니다. 인덱스를 지정하지 않으면, a.pop() 은 리스트의 마지막 항목을 삭제하고 돌려줍니다. (메서드 시그니처에서 i 를 둘러싼 대괄호는 매개변수가 선택적임을 나타냅니다. 그 위치에 대괄호를 입력해야 한다는 뜻이 아닙니다. 이 표기법은 파이썬 라이브러리 레퍼런스에서 지주 등장합니다.)

참고로, remove안에는 인자값으로 값자체를, pop은 인자값으로 인덱스를 넣기 때문에 pop을 활용해서 구현해야 한다. 이 부분 문서를 제대로 안 읽어서 조금 더 시간이 걸렸다. 

 

[2] 시간 제한 초과가 나면, input() 대신 sys.stdin.readline()을 활용하면 시간이 줄어든다.