[백준] 10828번 스택 - Python(파이썬)

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

10828번: 스택

 

문제

 

접근법

 

stack은 짐정리 개념이다. 즉, 나중에 들어온 애가 먼저 나가기.

파이썬의 list는 pop을 갖고 있으며 마지막으로 저장된 아이를 내보낸다.

즉, list는 stack구조로 사용하기 적합

 

 

코드

import sys

N=int(input())
stk=list()
for i in range(N):
    cmd=sys.stdin.readline().split()
    
    if cmd[0]=='push':
        stk.append(int(cmd[1]))
    elif cmd[0]=='pop':
        try:
            print(stk.pop())
        except:
            print(-1)
    elif cmd[0]=='size':
        print(len(stk))
    elif cmd[0]=='empty':
        if stk:
            print(0)
        else:
            print(1)
    else:
        try:
            print(stk[-1])
        except:
            print(-1)