[백준] 10845번 : 큐 python

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

10845번: 큐

 

문제 정보

큐

실버4

20m

O

설계 방법

queue class를 설계 후 코드 작성

코드

import sys
input = sys.stdin.readline

class Queue:
  def __init__(self):
    self.data = []
  def size(self):
    return len(self.data)
  def empty(self):
    if len(self.data) == 0:
      return 1
    else:
      return 0
  def push(self, x):
    self.data.append(x)
  def pop(self):
    if len(self.data) == 0:
      return -1
    else:
			# pop(0)으로 해도 같음 
      ans = self.data[0]
      self.data = self.data[1:]
      return ans
  def front(self):
    if len(self.data) == 0:
      return -1
    else:
      return self.data[0]
  def back(self):
    if len(self.data) == 0:
      return -1
    else:
      return self.data[-1]

n = int(input())
queue = Queue()
for _ in range(n):
  cmd = input().split()
  op = cmd[0]
  if op == 'size':
    print(queue.size())
  elif op == 'empty':
    print(queue.empty())
  elif op == 'push':
    queue.push(cmd[1])
  elif op == 'pop':
    print(queue.pop())
  elif op == 'front':
    print(queue.front())
  elif op == 'back':
    print(queue.back())
  else:
    print('err')

다른풀이

아래 코드와 같이 class를 구현하지 않고도 해결할 수 있음 

from sys import stdin

N = int(stdin.readline())
Que = []
for i in range(N) :
    A = stdin.readline().split()
    if A[0] == 'push' : Que.append(A[1])
    elif A[0] == 'pop' : 
        if Que : print(Que.pop(0))
        else : print(-1)
    elif A[0] == 'size' : print(len(Que))
    elif A[0] == 'empty' :
        if len(Que) == 0 : print(1)
        else : print(0)   
    elif A[0] == 'front' :
        if len(Que) == 0 : print(-1)
        else : print(Que[0])
    elif A[0] == 'back' :
        if len(Que) == 0 : print(-1)
        else : print(Que[-1])

시간 복잡도

$O(n)$

어려웠던 점

X

다만, python을 이용하는 가장 큰 이유가 내장함수 등이 잘 되어있다는 건데 그걸 잘 활용하지 못 하고 있다고 판단되었다,. python의 내장함수들을 조금 더 익혀야 할 필요를 느꼈다

참고 자료

백준 10845번 큐(Queue) 파이썬(Python)