Coding Test

[백준] 스택

상상쓰 2021. 7. 23. 16:18

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

 

10828번: 스택

첫째 줄에 주어지는 명령의 수 N (1 ≤ N ≤ 10,000)이 주어진다. 둘째 줄부터 N개의 줄에는 명령이 하나씩 주어진다. 주어지는 정수는 1보다 크거나 같고, 100,000보다 작거나 같다. 문제에 나와있지

www.acmicpc.net

 

스택은 Last In First Out!

 

import sys

N = int(sys.stdin.readline())
stack = []

for i in range(N):
    cmd = sys.stdin.readline().strip().split()
    
    if cmd[0] == 'push':
        stack.append(cmd[1])
    
    elif cmd[0] == 'pop':
        print(-1 if not stack else stack.pop())
    
    elif cmd[0] == 'size':
        print(len(stack))
    
    elif cmd[0] == 'empty':
        print(0 if stack else 1)
    
    else:
        print(-1 if not stack else stack[-1])