[python 구름] Stack

📜 접근 방법

1. push와 pop을 기준으로 나눈다.
2. push일 경우 stack의 크기 k와 현재 stack의 크기를 비교하여 다 찼으면 Overflow 출력
   1. 아닐 경우 값을 넣는다
3. pop일 경우 stack의 크기가 0일 경우 Underflow 출력
   1. 아닐 경우 pop을 한다.

❌ 실패 코드

 

✅ 정답 코드

import sys
input = sys.stdin.readline

n, k = map(int, input().split())
stack = []

for _ in range(n):
	command = list(input().split())
	if command[0] == 'push':
		if len(stack) == k:
			print('Overflow')
		else:
			stack.append(command[1])
	else:
		if len(stack) == 0:
			print('Underflow')
		else:
			print(stack.pop())

💡알게 된점