병훈's Blog

[Python] [백준] 10866번 - 덱 (deque 사용) 본문

Algorithm

[Python] [백준] 10866번 - 덱 (deque 사용)

thdqudgns 2022. 10. 10. 02:25

정수를 저장하는 덱(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보다 작거나 같다. 문제에 나와있지 않은 명령이 주어지는 경우는 없다.

출력

출력해야하는 명령이 주어질 때마다, 한 줄에 하나씩 출력한다.

예제 입력 1 복사

15
push_back 1
push_front 2
front
back
size
empty
pop_front
pop_back
pop_front
size
empty
pop_back
push_front 3
empty
front

예제 출력 1 복사

2
1
2
0
2
1
-1
0
1
-1
0
3

from collections import deque
import sys

N = int(input())
deque = deque()

for _ in range(N):
    data = list(map(str, sys.stdin.readline().strip().split(' ')))
    com = data[0]
    if len(data) > 1:
        num = int(data[1])

    if com == "push_front":
        deque.appendleft(num)
    elif com == "push_back":
        deque.append(num)
    elif com == "pop_front":
        if len(deque) == 0:
            print(-1)
        else:
            print(deque.popleft())
    elif com == "pop_back":
        if len(deque) == 0:
            print(-1)
        else:
            print(deque.pop())
    elif com == "size":
        print(len(deque))
    elif com == "empty":
        if len(deque) == 0:
            print(1)
        else:
            print(0)
    elif com == "front":
        if len(deque) == 0:
            print(-1)
        else:
            print(deque[0])
    elif com == "back":
        if len(deque) == 0:
            print(-1)
        else:
            print(deque[-1])

딱히 설명할 게 없는 문제다

from collections import deque

로 deque를 import하고,

 

data = list(map(str, sys.stdin.readline().strip().split(' ')))
com = data[0]
if len(data) > 1:
    num = int(data[1])

로 명령어와 정수를 각각 com과 num에 저장한다.

그리고 list에서 사용하는 명령어들로 구현하면 된다.

 

차이점은 두 가지 뿐이다.

deque.appendleft(num)
print(deque.popleft())

먼저 입력받은 것들은 왼쪽에 위치하는데

deque의 특성상 양쪽에서 입력, 출력이 가능하기 때문에

appendleft, popleft 이 두 개는 

가장 먼저 입력받은 것처럼 추가하거나

가장 먼저 입력받은 것을 뽑아낼 수 있다.

728x90
728x90