파이톨치

[백준] 18258 - 큐2 본문

알고리즘

[백준] 18258 - 큐2

파이톨치 2021. 9. 24. 16:01
728x90

[문제 출처]

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

 

18258번: 큐 2

첫째 줄에 주어지는 명령의 수 N (1 ≤ N ≤ 2,000,000)이 주어진다. 둘째 줄부터 N개의 줄에는 명령이 하나씩 주어진다. 주어지는 정수는 1보다 크거나 같고, 100,000보다 작거나 같다. 문제에 나와있지

www.acmicpc.net

[어떻게 풀까?]

처음엔 굉장히 간단하게 생각했다. 문제 구현이 복잡한 것도 아니고 어려운 알고리즘을 구성하는 것도 아니었기 때문이다. 하지만 당연하게도 시간초과가 뜨고 말았다. 왜 그런가 생각해 봤다. 이 문제는 큐에 관한 문제이다. 하지만 큐도 사용하지 않고 문제를 풀려고 했기 때문에 시간초과가 뜬 것은 어찌보면 당연한 일이다. 

파이썬에서 큐에 대해 사용하는 것은 

https://leonkong.cc/posts/python-deque.html

 

Python - 데크(deque) 언제, 왜 사용해야 하는가?

Python의 데크(deque)에 대해 알아보고 언제, 왜 써야 하는지 살펴본다

leonkong.cc

아래의 블로그를 참고하면 된다.

[코드]

ㅇㄹㄴㅇ
from collections import deque
import sys

num = int(input())
array = deque([])
for i in range(num):
  temp = sys.stdin.readline().split()
  if temp[0] == 'push':
    array.append(temp[1])
  elif temp[0] == 'front':
    if len(array) == 0:
      print(-1)
    else:
      print(array[0])
  elif temp[0] == 'pop':
    if len(array) == 0:
      print(-1)
    else:
      print(array.popleft())
  elif temp[0] == 'size':
    print(len(array))
  elif temp[0] == 'empty':
    if not array:
      print(1)
    else:
      print(0)
  elif temp[0] == 'back':
    if not array:
      print(-1)
    else:
      print(array[-1])
728x90