[BOJ | Python] 28279번: 덱 2

https://www.acmicpc.net/problem/28279

 

문제

정수를 저장하는 덱을 구현한 다음, 입력으로 주어지는 명령을 처리하는 프로그램을 작성하시오.

명령은 총 여덟 가지이다.

1. 1 X: 정수 X를 덱의 앞에 넣는다. (1 ≤ X ≤ 100,000)
2. 2 X: 정수 X를 덱의 뒤에 넣는다. (1 ≤ X ≤ 100,000)
3. 3: 덱에 정수가 있다면 맨 앞의 정수를 빼고 출력한다. 없다면 -1을 대신 출력한다.
4. 4: 덱에 정수가 있다면 맨 뒤의 정수를 빼고 출력한다. 없다면 -1을 대신 출력한다.
5. 5: 덱에 들어있는 정수의 개수를 출력한다.
6. 6: 덱이 비어있으면 1, 아니면 0을 출력한다.
7. 7: 덱에 정수가 있다면 맨 앞의 정수를 출력한다. 없다면 -1을 대신 출력한다.
8. 8: 덱에 정수가 있다면 맨 뒤의 정수를 출력한다. 없다면 -1을 대신 출력한다.

 

입력

첫째 줄에 명령의 수 N이 주어진다. (1 ≤ N ≤ 1,000,000)

둘째 줄부터 N개 줄에 명령이 하나씩 주어진다.

출력을 요구하는 명령은 하나 이상 주어진다.

 

 

출력

출력을 요구하는 명령이 주어질 때마다 명령의 결과를 한 줄에 하나씩 출력한다.

 

 

풀이

import sys
from collections import deque

input = sys.stdin.readline
N = int(input())
D = deque()
ans = []

for _ in range(N):
    command = list(map(int, input().split()))

    if command[0] == 1:
        D.appendleft(command[1])
    elif command[0] == 2:
        D.append(command[1])
    elif command[0] == 3:
        ans.append(str(D.popleft() if D else -1))
    elif command[0] == 4:
        ans.append(str(D.pop() if D else -1))
    elif command[0] == 5:
        ans.append(str(len(D)))
    elif command[0] == 6:
        ans.append(str(1 if not D else 0))
    elif command[0] == 7:
        ans.append(str(D[0] if D else -1))
    elif command[0] == 8:
        ans.append(str(D[-1] if D else -1))

print("\n".join(ans))

 

python의 deque를 사용하였다.

처음에는 빠른 입출력을 사용하지 않고 하나씩 입력받고 각각 출력하는 방식을 사용했는데, 이렇게 하니 계속 시간초과가 발생했다,,, 

빠른 입출력의 중요성을 다시 한 번 깨닫게 되었다.