PS/큐(queue)

[백준 파이썬(python) 10845번 문제 큐 ]

이거시원조랑께 2023. 8. 14. 07:13
반응형

문제

설명

하라는 대로 if문 오지게 따라치면 되는 간단한 문제

코드

import sys
from collections import deque

dq = deque()
N = int(sys.stdin.readline())

for _ in range(N):
    command = sys.stdin.readline().split()
    if command[0] == "push":
        dq.append(int(command[1]))
    if command[0] == "pop":
        if len(dq) == 0:
            print(-1)
        else:
            print(dq.popleft())
    if command[0] == "size":
        print(len(dq))
    if command[0] == "empty":
        if len(dq) == 0:
            print(1)
        else:
            print(0)
    if command[0] == "front":
        if len(dq) == 0:
            print(-1)
        else:
            print(dq[0])
    if command[0] == "back":
        if len(dq) == 0:
            print(-1)
        else:
            print(dq[-1])

코드가 길어서 짧게 쓴 다른 사람꺼 봤는데 처음 보는 'lamda' 를 쓰길래 빠르게 포기했다. 큐큐..

반응형