덱
- 시간 제한 : 0.5초
- 메모리 제한 : 256MB
알고리즘 분류
- 구현
- 자료 구조
- 덱
문제
정수를 저장하는 덱(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보다 작거나 같다. 문제에 나와있지 않은 명령이 주어지는 경우는 없다.
출력
출력해야하는 명령이 주어질 때마다, 한 줄에 하나씩 출력한다.
풀이
시간 제한이 0.5초이기 때문에 python에서 빠른 입력을 받기 위해 sys.stdin.readline().rstrip()을 사용했다. 각 명령마다 아래와 같은 함수를 사용하였다.
- push_front X: 리스트에 appendleft() 메소드를 이용
- push_back X: 리스트에 append() 메소드를 이용
- pop_front: 리스트에 popleft() 메소드를 이용
- pop_back: 리스트에 pop() 메소드를 이용
- size: len() 메소드를 이용
- empty, front, back: 리스트 유무를 따지는 조건문을 이용
덱
스택과 큐를 일반화한 것으로 LIFO(Last In First Out) 방식의 스택으로 사용할 수 있고 FIFO(First In First Out) 방식의 큐로도 사용할 수 있다..
양방향으로 append와 pop연산이 가능하며, 시간 복잡도 O(1) 성능을 지원하기 때문에 O(N)인 리스트보다 성능이 좋다.
덱은 deque 모듈을 이용해 구현할 수 있다.
소스 코드
Python
import sys
from collections import deque
N = int(sys.stdin.readline().rstrip())
array = deque([])
for i in range(N):
command = sys.stdin.readline().rstrip().split()
if (command[0] == "push_front"):
array.appendleft(command[1])
elif (command[0] == "push_back"):
array.append(command[1])
elif (command[0] == "pop_front"):
if array:
print(array.popleft())
else:
print("-1")
elif (command[0] == "pop_back"):
if array:
print(array.pop())
else:
print("-1")
elif (command[0] == "size"):
print(len(array))
elif (command[0] == "empty"):
if array:
print("0")
else:
print("1")
elif (command[0] == "front"):
if array:
print(array[0])
else:
print("-1")
else:
if array:
print(array[len(array) - 1])
else:
print("-1")