[백준] 18258번: 큐2_실버4 (Python)

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

18258번: 큐 2

문제

정수를 저장하는 큐를 구현한 다음, 입력으로 주어지는 명령을 처리하는 프로그램을 작성하시오.

명령은 총 여섯 가지이다.

• push X: 정수 X를 큐에 넣는 연산이다.
• pop: 큐에서 가장 앞에 있는 정수를 빼고, 그 수를 출력한다. 만약 큐에 들어있는 정수가 없는 경우에는 -1을 출력한다.
• size: 큐에 들어있는 정수의 개수를 출력한다.
• empty: 큐가 비어있으면 1, 아니면 0을 출력한다.
• front: 큐의 가장 앞에 있는 정수를 출력한다. 만약 큐에 들어있는 정수가 없는 경우에는 -1을 출력한다.
• back: 큐의 가장 뒤에 있는 정수를 출력한다. 만약 큐에 들어있는 정수가 없는 경우에는 -1을 출력한다.

입력

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

출력

출력해야하는 명령이 주어질 때마다, 한 줄에 하나씩 출력한다.

예제 입력 1 복사

15
push 1
push 2
front
back
size
empty
pop
pop
pop
size
empty
pop
push 3
empty
front

예제 출력 1 복사

1
2
2
0
1
2
-1
0
1
-1
0
3

출처

• 문제를 만든 사람: jh05013

알고리즘 분류

• 자료 구조
• 큐

시간 제한

• Python 3: 3 초
• PyPy3: 3 초
• Python 2: 3 초
• PyPy2: 3 초

---

import sys

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

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

    if(command[0] == "push"):
        queue_.append(int(command[1]))

    if (command[0] == "pop"):
        if(len(queue_) == 0):
            print("-1")
        else:
            print(queue_.pop(0))

    if (command[0] == "size"):
        print(len(queue_))

    if (command[0] == "empty"):
        if (len(queue_) == 0):
            print("1")
        else: print("0")

    if (command[0] == "front"):
        if(len(queue_) == 0):
            print("-1")
        else: print(queue_[0])

    if (command[0] == "back"):
        if (len(queue_) == 0):
            print("-1")
        else:
            print(queue_[len(queue_) - 1])

 

위 코드는 큐1 문제 소스코드이다. 큐1과 문제는 같지만 시간 제한이 다르다.

 

큐2에서는 시간 초과가 떠서 어떻게하면 시간복잡도를 줄일 수 있나 찾아봤더니

deque을 사용하면 list를 사용하여 queue를 구현한 것보다 시간 복잡도를 줄일 수 있는 것이었다.

 

import sys
from collections import deque

n = int(sys.stdin.readline())
queue_ = deque()

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

    if(command[0] == "push"):
        queue_.append(int(command[1]))

    if (command[0] == "pop"):
        if(len(queue_) == 0):
            print("-1")
        else:
            print(queue_.popleft())

    if (command[0] == "size"):
        print(len(queue_))

    if (command[0] == "empty"):
        if (len(queue_) == 0):
            print("1")
        else: print("0")

    if (command[0] == "front"):
        if(len(queue_) == 0):
            print("-1")
        else: print(queue_[0])

    if (command[0] == "back"):
        if (len(queue_) == 0):
            print("-1")
        else:
            print(queue_[len(queue_) - 1])

 

 

리스트를 이용하면 맨 앞의 원소를 삭제하기 위해서 pop(0)을 했을 때

빈 자리를 채우기 위해 원소들이 하나하나 앞으로 움직이기 때문에 시간 복잡도가 O(n)이지만 

 

덱은 하나하나가 이중 연결 리스트의 객체로 돼있기 때문에 시간복잡도가 O(0)이 된다.