[🥈4 / 백준 10866 / 파이썬] 덱 (deque 라이브러리 사용 x)

 

10866번: 덱

 

문제

정수를 저장하는 덱(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
예제 입력 2  22 front back pop_front pop_back push_front 1 front pop_back push_back 2 back pop_front push_front 10 push_front 333 front back pop_back pop_back push_back 20 push_back 1234 front back pop_back pop_back	예제 출력 2  -1 -1 -1 -1 1 1 2 2 333 10 10 333 20 1234 1234 20

 

풀이

 

파이썬의 deque 라이브러리 사용 없이 덱을 구현해보았다.

 

덱은 큐나 스택과 달리 양방향에서 입출력이 가능한 자료구조이다.

 

pop이나 append등의 함수를 써서도 구현이 가능하겠지만, 이 경우 시간복잡도가 O(n)이 되어버린다.

 

가급적 O(1)에 가깝게 구현하기 위해 최대 명령 입력 수인 10,000 길이의 배열을 만들어 주고 front와 rear 인덱스를 활용했다.

 

import sys
input = sys.stdin.readline
deque = [0] * 10000
f,r = 0,0
for i in range(int(input())):
    a = input().split()
    if a[0] == "push_front":
        deque[f] = a[1]
        f+=1
    elif a[0] == "push_back":
        r -= 1
        deque[r] = a[1]
        # front와 back의 인덱스 증가 순서에 유의해야 한다.
        # 둘이 겹치면 안되므로 front는 자료를 바꾼 뒤 인덱스 변경,
        # back은 인덱스를 먼저 바꾼 후 자료를 입력한다.
    elif a[0] == "pop_front":
        if f-r:
            f -= 1
            print(deque[f])
        else:
            print(-1)
    elif a[0] == "pop_back":
        if f-r:
            print(deque[r])
            r += 1
        else:
            print(-1)
    elif a[0] == "size":
        print(f-r)
        # f와 r의 위치가 같다면 비어있는 것이다.
    elif a[0] == "empty":
        print(1 if f-r == 0 else 0)
    elif a[0] == "front":
        print(deque[f-1] if f-r else -1)
    elif a[0] == "back":
        print(deque[r] if f-r else -1)