10828번: 스택
첫째 줄에 주어지는 명령의 수 N (1 ≤ N ≤ 10,000)이 주어진다. 둘째 줄부터 N개의 줄에는 명령이 하나씩 주어진다. 주어지는 정수는 1보다 크거나 같고, 100,000보다 작거나 같다. 문제에 나와있지
www.acmicpc.net
본 코드는 Python으로 작성되었습니다.
기본적인 스택을 사용하는 예제로, 자료구조에 알고리즘 이해를 늘일 수 있게 되었습니다.
import sys
n = int(sys.stdin.readline())
stack = []
for i in range(n):
command = sys.stdin.readline().split()
if command[0] == "push":
stack.append(command[1])
elif command[0] == "pop":
if not stack:
print(-1)
else:
print(stack.pop())
elif command[0] == "size":
print(len(stack))
elif command[0] == "empty":
if not stack:
print(1)
else:
print(0)
elif command[0] == "top":
if not stack:
print(-1)
else:
print(stack[len(stack) - 1])
728x90
'코딩테스트 기록 > 03. Stack' 카테고리의 다른 글
| [백준] 4949: 균형잡힌 세상 (Pyhton) (1) | 2023.11.13 |
|---|