[Python] 백준 #18258 - 큐 2

문제

---

18258번: 큐 2

코드

---

My answer

import sys
from collections import deque

input=sys.stdin.readline

n=int(input())
que=deque()
for i in range(n):
    m=list(input().split())
    if(m[0]=='push'):
        que.append(int(m[1]))
    elif(m[0]=='pop'):
        if(que):
            print(que.popleft())
        else:print(-1)
    elif(m[0]=='size'):
        print(len(que))
    elif(m[0]=='empty'):
        if(que):print(0)
        else:print(1)
    elif(m[0]=="front"):
        if(que):print(que[0])
        else:print(-1)
    else:
        if(que):print(que[-1])
        else:print(-1)

Another answer

from collections import*
q,p=deque(),print
def u(z):q.append(z)
def o(z):p(q.popleft()if q else-1)
def i(z):p(len(q))
def m(z):p(+(not q))
def r(z):p(q[0]if q else-1)
def a(z):p(q[-1]if q else-1)
for s in[*open(0)][1:]:{'u':u,'o':o,'i':i,'m':m,'r':r,'a':a}[s[1]](s[5:-1])

풀이

---

다른 명령들은 그냥 수행하면되는데 pop같은 경우에는 파이썬의 기본pop을 쓰게 되면 맨 뒤에껄 pop하는건 문제가 안되지만 맨 앞의 원소를 pop하면 전부 땡겨줘야한다. 이 땡겨주는 부분에서 시간초과가 발생하기 때문에 collections모듈의 deque함수를 사용해서 리스트가 아닌 deuqe구조로 만든후 popleft를 쓰면 해결된다. 다음 시간에 알고리즘 게시판에 deque에 대해서 작성해보도록 하겠다.