[BaekJoon] 10845 큐

[출처 : https://www.acmicpc.net/problem/10845]

 

▶ 자료구조 큐(Queue)를 구현하는 문제이다.

import sys

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

q = list()

while n > 0:
    cmd = sys.stdin.readline().rstrip()
    ## .rstrip() 를 안붙여 주면 입력값이 예를들어 "push 1" 인경우 cmd에는 "push 1\n" 으로 개행 문자열까지 포함된다. 따라서 .rstrip()를 붙여 올바른 입력을 받도록한다.

    val = ''

    if ' ' in cmd:
        cmd, val = cmd.split()
        val = int(val)
       
    if cmd == 'push':
        q.append(val)
    elif cmd == 'pop':
        if len(q) == 0:
            sys.stdout.write(str(-1)+'\n')
            ## write함수는 스트링 값만 출력 할 수 있다. print() 처럼 ..write(1) 이 불가능하다.
        else:    
            sys.stdout.write(str(q[0])+'\n')
            del q[0]
    elif cmd == 'size':
         sys.stdout.write(str(len(q))+'\n')
    elif cmd == 'empty':
        if len(q) == 0:
             sys.stdout.write(str(1)+'\n')
        else:
             sys.stdout.write(str(0)+'\n')
    elif cmd == 'front':
        if len(q) == 0:
             sys.stdout.write(str(-1)+'\n')
        else:
             sys.stdout.write(str(q[0])+'\n')
    elif cmd == 'back':
        if len(q) == 0:
             sys.stdout.write(str(-1)+'\n')
        else:
             sys.stdout.write(str(q[len(q)-1])+'\n')

    n -= 1

풀이과정

1. 큐로 사용하기 위한 리스트를 선언

2. 입력을 받아 명령과 값으로 문자열 파싱

 

3. 각각의 명령에 맞는 동작 정의

 

 

★★★ 초기 시간초과 오류가 발생하였는데 이는 파이썬 빠른 입출력을 이용해 해결 할 수 있었다.

+) 추가적으로 파이썬 빠른 입출력 사용법

 

sys.stdin.readline()은 뒤에 개행문자("\n")도 받아들이기 때문에 이런 경우 sys.stdin.readline().rstrip() 로 사용해야 개행문자를 무시한다.

 

한편 출력에 해당하는 sys.stdout.write() 은 개행처리를 별도로 해줘야 한다. Ex) sys.stdout.write(str(0)+'\n')