[백준]10866번 덱 - 덱

 

from collections import deque
import sys

n = int(sys.stdin.readline())
deque = deque([])
for i in range(n):
    command = sys.stdin.readline().strip()
    if command.split()[0] == 'push_front':
        deque.appendleft(int(command.split()[1]))
    elif command.split()[0] == 'push_back':
        deque.append(int(command.split()[1]))
    elif command == 'pop_front':
        if len(deque) == 0:
            print(-1)
        else:
            print(deque.popleft())
    elif command == 'pop_back':
        if len(deque) == 0:
            print(-1)
        else:
            print(deque.pop())
    elif command == 'size':
        print(len(deque))
    elif command == 'empty':
        if len(deque) == 0:
            print(1)
        else:
            print(0)
    elif command == 'front':
        if len(deque) == 0:
            print(-1)
        else:
            print(deque[0])
    elif command == 'back':
        if len(deque) == 0:
            print(-1)
        else:
            print(deque[-1])

 

코드 설명

덱 사용을 위해 collections 모듈의 deque 함수를 import 한다. 명령어를 입력 받으면서 각 명령의 기능에 맞게 함수를 구성한다.

front X : appendleft(), back x : append(), pop_front : popleft(), pop_back : pop(), empty : 덱의 길이로 체크, front : print(deque[0]), back : print(deque[-1])

 

Point : 명령어의 인수가 2개일 경우 command.split()[0] 으로 명령어, [1]을 통해 값을 확인할 수 있다.