백준 - 10845번(큐)

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

10845번: 큐

 

import sys
from collections import deque

n = int(sys.stdin.readline())

arr = deque([]) #덱 만들어주기

for _ in range(n):
    x = list(sys.stdin.readline().split()) #띄어쓰기를 기준으로 문자열을 원소로 할당
    
    if x[0] == 'push':
        arr.append(int(x[1]))

    elif x[0] == 'pop':
        if len(arr) == 0:
            print(-1)
        else:
            print(arr.popleft())

    elif x[0] == 'size':
        print(len(arr))

    elif x[0] == 'empty':
        if len(arr) == 0:
            print(1)
        else:
            print(0)

    elif x[0] == 'front':
        if len(arr) == 0:
            print(-1)
        else:
            t = arr.popleft()
            print(t)
            arr.appendleft(t)

    elif x[0] == 'back':
        if len(arr) == 0:
            print(-1)
        else:
            t = arr.pop()
            print(t)
            arr.append(t)

해당 명령어가 큐의 어떤 함수에 할당하는지 생각하면 쉽게 구현 가능하다.

stack의 명령어를 구현하는 문제와 매우 유사하다.

arr의 길이가 0일 때와 시간 초과가 나지 않기 위해 sys모듈을 사용한다는 것을 신경 써주면 된다.