[파이썬] 10845번: 큐

10845번: 큐 (acmicpc.net)

10845번: 큐

 

분명 맞게 푼 것 같았는데 답과는 다르게 출력돼서 뭐지..? 했는데 queue를 for문 내에 정의한 걸 디버깅하다가 깨달았다^^

그리고 pop을 수행할 때에도 리턴한 값을 출력하기 위해서 print()를 씌워줘야 한다. 

비슷한 스택 문제를 풀어봤어서 비교적 수월하게 풀 수 있었다. 

 

이 문제도 역시 n의 입력이 1부터 10000까지 주어지기 때문에 빠른 처리를 위해서 sys 모듈을 사용해야 한다.

n도 명령도 전부 sys로 입력받아야 한다! 

 

코드

import sys

n = int(sys.stdin.readline())
queue = []

for i in range(n):
    lst = sys.stdin.readline().split()

    if lst[0] == 'push':
        queue.append(int(lst[1]))
    elif lst[0] == 'pop':
        if len(queue) != 0:
            print(queue.pop(0))
        else:
            print(-1)
    elif lst[0] == 'size':
        print(len(queue))
    elif lst[0] == 'empty':
        if len(queue) == 0:
            print(1)
        else:
            print(0)
    elif lst[0] == 'front':
        if len(queue) == 0:
            print(-1)
        else:
            print(queue[0])
    elif lst[0] == 'back':
        if len(queue) == 0:
            print(-1)
        else:
            print(queue[-1])