코딩테스트/백준
[백준] 10828번 스택 / 파이썬(python)
Jiwon_C
2021. 3. 23. 23:23
# 문제 링크
10828번: 스택
첫째 줄에 주어지는 명령의 수 N (1 ≤ N ≤ 10,000)이 주어진다. 둘째 줄부터 N개의 줄에는 명령이 하나씩 주어진다. 주어지는 정수는 1보다 크거나 같고, 100,000보다 작거나 같다. 문제에 나와있지
www.acmicpc.net
# Soultion
시간초과 문제가 발생하므로, 5줄에서 input() 대신 sys.stdin.readline()을 이용하여 읽어온다.파이썬에서는 리스트를 이용하여 스택으로 사용이 가능하다.
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
|
import sys
n = int(input())
stack = []
for _ in range(n):
cmd = list(map(str,sys.stdin.readline().split()))
if cmd[0]=='push':
stack.append(int(cmd[1]))
elif cmd[0]=='pop':
if stack:
print(stack.pop())
else:
print("-1")
elif cmd[0]=='size':
print(len(stack))
elif cmd[0]=='empty':
if len(stack)==0:
print(1)
else:
print(0)
elif cmd[0]=='top':
if stack:
print(stack[-1])
else:
print("-1")
|
cs |