덱(백준 10866번)
문제
정수를 저장하는 덱(Deque)를 구현한 다음, 입력으로 주어지는 명령을 처리하는 프로그램을 작성하시오.
명령은 총 여덟 가지이다.
- push_front X: 정수 X를 덱의 앞에 넣는다.
- push_back X: 정수 X를 덱의 뒤에 넣는다.
- pop_front: 덱의 가장 앞에 있는 수를 빼고, 그 수를 출력한다. 만약, 덱에 들어있는 정수가 없는 경우에는 -1을 출력한다.
- pop_back: 덱의 가장 뒤에 있는 수를 빼고, 그 수를 출력한다. 만약, 덱에 들어있는 정수가 없는 경우에는 -1을 출력한다.
- size: 덱에 들어있는 정수의 개수를 출력한다.
- empty: 덱이 비어있으면 1을, 아니면 0을 출력한다.
- front: 덱의 가장 앞에 있는 정수를 출력한다. 만약 덱에 들어있는 정수가 없는 경우에는 -1을 출력한다.
- back: 덱의 가장 뒤에 있는 정수를 출력한다. 만약 덱에 들어있는 정수가 없는 경우에는 -1을 출력한다.
나의 풀이
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
import sys
from collections import deque
N=int(input())
deq=deque()
for _ in range(N):
com=sys.stdin.readline().strip()
c=com.split()[0]
if c=='push_front':
deq.appendleft(com.split()[-1])
elif c=='push_back':
deq.append(com.split()[-1])
elif c=='pop_front':
if deq:
print(deq.popleft())
else:
print(-1)
elif c=='pop_back':
if deq:
print(deq.pop())
else:
print(-1)
elif c=='size':
print(len(deq))
elif c=='empty':
if deq:
print(0)
else:
print(1)
elif c=='front':
if deq:
print(deq[0])
else:
print(-1)
elif c=='back':
if deq:
print(deq[-1])
else:
print(-1)
- 파이썬은
collections
라이브러리에deque()
가 구현되어 있기 때문에, 쉽게 풀 수 있었다.
This post is licensed under CC BY 4.0 by the author.