10845번 : 큐 - Python

문제

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

풀이

list로 큐를 구현할 수 있다. pop은 pop(0)으로 가장 앞에 있는 수를 빼면서 출력할 수 있다. back 같은 경우도 list[-1]로 출력할 수 있다.

코드

from sys import stdin

input = stdin.readline

def push(array, n):
    array.append(n)

def pop(array):
    if array:
        return array.pop(0)
    else:
        return -1

def size(array):
    return len(array)

def empty(array):
    if array:
        return 0
    else:
        return 1

def front(array):
    if array:
        return array[0]
    else:
        return -1

def back(array):
    if array:
        return array[-1]
    else:
        return -1

n = int(input())
array = list()

for i in range(n):
    
    m = input().split()

    if 'push' in m:
        push(array,int(m[-1]))
    elif 'pop' in m:
        print(pop(array))
    elif 'size' in m:
        print(size(array))
    elif 'empty' in m:
        print(empty(array))
    elif 'front' in m:
        print(front(array))
    elif 'back' in m:
        print(back(array))