BOJ 18258 : 큐 2 (Python)

문제

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

18258번: 큐 2

 

 

코드

import sys

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

que = [0] * 2000001
front = 0
rear = 0
no = 0

def is_empty():
    global no
    if no<=0:
        return 1
    else: return 0

def size():
    global no
    return no

def push(num):
    global rear
    global no
    que[rear] = num
    rear += 1
    no += 1
    if rear == 2000001:
        rear = 0

def pop():
    global no
    global front
    if no<=0:
        return -1
    else:
        x = que[front]
        front += 1
        no -= 1
        if front == 2000001:
            front = 0
        return x

def front_check():
    global front
    if is_empty()==1:
        return -1
    else: return que[front]

def back():
    global rear
    if is_empty()==1:
        return -1
    elif rear == 0:
        return que[2000001-1]
    
    else : return que[rear-1]

for i in range(N):
    a = sys.stdin.readline().rstrip()
    b = ''
    if a == "front":
        print(front_check())
    elif a == "back":
        print(back())
    elif a == "size":
        print(size())
    elif a == "empty":
        print(is_empty())
    elif a == "pop":
        print(pop())
    else:
        a,b = a.split()
        push(int(b))

 

 

풀이

 

그냥 큐 기본 함수 쓰는 문제...

근데 python3 기준으로 그냥 input() 사용 시 시간초과가 뜬다. 

그래서 input을 sys.stdin.readline().rstrip() 으로 바꿔서 풀면 시간 초과가 해결된다.

이것때문인지도 모르고 다른 부분만 계속 고쳤다.....