[BOJ/백준] 10866 - 덱 (Python)

문제

문제링크

주어진 명령은 총 8개이다.

• push_front X: 정수 X를 덱의 앞에 넣는다.
• push_back X: 정수 X를 덱의 뒤에 넣는다.
• pop_front: 덱의 가장 앞에 있는 수를 빼고, 그 수를 출력한다. 만약, 덱에 들어있는 정수가 없는 경우에는 -1을 출력한다.
• pop_back: 덱의 가장 뒤에 있는 수를 빼고, 그 수를 출력한다. 만약, 덱에 들어있는 정수가 없는 경우에는 -1을 출력한다.
• size: 덱에 들어있는 정수의 개수를 출력한다.
• empty: 덱이 비어있으면 1을, 아니면 0을 출력한다.
• front: 덱의 가장 앞에 있는 정수를 출력한다. 만약 덱에 들어있는 정수가 없는 경우에는 -1을 출력한다.
• back: 덱의 가장 뒤에 있는 정수를 출력한다. 만약 덱에 들어있는 정수가 없는 경우에는 -1을 출력한다.

n개의 줄에 명령이 하나씩 주어질 때, 명령을 실행하라.

 

풀이

명령어를 공백 기준으로 나눠 리스트 형태로 저장할 때, push_front X와 push_back X만 두 개 요소가 저장된다.

따라서 리스트의 길이가 2면 뒤 요소를 int형으로 바꾸고 각 명령어에 맞춰 dq에 집어넣으면 된다.append()는 맨 오른쪽에, appendleft()는 맨 왼쪽에 집어넣는다.그외의 경우, 즉 명령어가 하나로 이루어진 경우 empty와 size 명령어를 먼저 처리한다.empty는 비었는지 아닌지에 따라 if문으로 처리해 각각 1과 0을 출력한다.size는 단순히 len()으로 길이를 출력하면 된다.나머지 명령어에서 dq가 비어있으면 모두 -1을 출력해야 하므로 if문으로 먼저 구현한다.그 후 pop_front 명령어에서 popleft()로 맨 왼쪽(맨 앞) 값을 꺼내 출력하고 pop_back 명령어에서는 pop()으로 맨 오른쪽(맨 뒤) 값을 꺼내 출력한다.front와 back 명령어에서는 인덱스를 활용해 해당 위치의 요소를 출력하기만 하면 된다.  

 

코드

import sys
input=sys.stdin.readline
from collections import deque

dq=deque([])
def command():
    c=input().split()
    if len(c)==2:
        x=int(c[-1])
        if c[0]=='push_front':
            dq.appendleft(x)
        elif c[0]=='push_back':
            dq.append(x)
    else:
        if c[0]=='empty':
            if dq:
                print(0)
            else:
                print(1)
        elif c[0]=='size':
            print(len(dq))
        else:
            if not dq:
                print(-1)
            elif c[0]=='pop_front':
                print(dq.popleft())
            elif c[0]=='pop_back':
                print(dq.pop())
            elif c[0]=='front':
                print(dq[0])
            elif c[0]=='back':
                print(dq[-1])

n=int(input())
for _ in range(n):
    command()