반응형
파이썬 덱을 구현해 보는 기회였다. 시간 초과때문에 빡쳤지만, sys.stdin.readline() 에서 split(" ") 가 아닌 split() 만 작동함을 깨우쳤다.
문제
정수를 저장하는 덱(Deque)를 구현한 다음, 입력으로 주어지는 명령을 처리하는 프로그램을 작성하시오.
명령은 총 여덟 가지이다.
- push_front X: 정수 X를 덱의 앞에 넣는다.
- push_back X: 정수 X를 덱의 뒤에 넣는다.
- pop_front: 덱의 가장 앞에 있는 수를 빼고, 그 수를 출력한다. 만약, 덱에 들어있는 정수가 없는 경우에는 -1을 출력한다.
- pop_back: 덱의 가장 뒤에 있는 수를 빼고, 그 수를 출력한다. 만약, 덱에 들어있는 정수가 없는 경우에는 -1을 출력한다.
- size: 덱에 들어있는 정수의 개수를 출력한다.
- empty: 덱이 비어있으면 1을, 아니면 0을 출력한다.
- front: 덱의 가장 앞에 있는 정수를 출력한다. 만약 덱에 들어있는 정수가 없는 경우에는 -1을 출력한다.
- back: 덱의 가장 뒤에 있는 정수를 출력한다. 만약 덱에 들어있는 정수가 없는 경우에는 -1을 출력한다.
입력
첫째 줄에 주어지는 명령의 수 N (1 ≤ N ≤ 10,000)이 주어진다. 둘째 줄부터 N개의 줄에는 명령이 하나씩 주어진다. 주어지는 정수는 1보다 크거나 같고, 100,000보다 작거나 같다. 문제에 나와있지 않은 명령이 주어지는 경우는 없다.
출력
출력해야하는 명령이 주어질 때마다, 한 줄에 하나씩 출력한다.
import collections
import sys
N = int(sys.stdin.readline())
dec= collections.deque([])
for _ in range(N):
commands = list(sys.stdin.readline().split()) # sys.stdin.readline() 에서 split 은 " "가 작동하지 않는다.
if commands[0]=="push_back":
dec.append(int(commands[1]))
continue
elif commands[0]=="push_front":
dec.appendleft(int(commands[1]))
continue
elif commands[0]=="pop_back":
if len(dec)>0:
print(dec.pop())
else:
print(-1)
continue
elif commands[0]=="pop_front":
if len(dec)>0:
print(dec.popleft())
else:
print(-1)
continue
elif commands[0]=="front":
if len(dec)>0:
print(dec[0])
else:
print(-1)
continue
elif commands[0]=="back":
if len(dec)>0:
print(dec[-1])
else:
print(-1)
continue
elif commands[0]=="size":
print(len(dec))
continue
elif commands[0]=="empty":
print(int(len(dec)==0))
continue
반응형
'알고리즘' 카테고리의 다른 글
백준 9375번 패션왕 신혜빈 파이썬 (0) | 2020.09.30 |
---|---|
백준 10250번 ACM 호텔 파이썬 (0) | 2020.09.30 |
백준 11050번 이항계수 1 파이썬 (0) | 2020.09.30 |
백준 11051번 이항계수 2 파이썬 (0) | 2020.09.30 |
백준 11279번 최대 힙 파이썬 (0) | 2020.09.30 |