[백준][실버] 24511번 queuestack

📁문제 설명

한가롭게 방학에 놀고 있던 도현이는 갑자기 재밌는 자료구조를 생각해냈다. 그 자료구조의 이름은 queuestack이다.

queuestack의 구조는 다음과 같다.
$1$번,
$2$번, ... ,
$N$번의 자료구조(queue 혹은 stack)가 나열되어있으며, 각각의 자료구조에는 한 개의 원소가 들어있다.

queuestack의 작동은 다음과 같다.

• b$x_0$을 입력받는다.

• $x_0$을 $1$번 자료구조에 삽입한 뒤 $1$번 자료구조에서 원소를 pop한다. 그때 pop된 원소를 $x_1$이라 한다.

• $x_1$을 $2$번 자료구조에 삽입한 뒤 $2$번 자료구조에서 원소를 pop한다. 그때 pop된 원소를 $x_2$이라 한다.

• ...

• $x_{N-1}$을 $N$번 자료구조에 삽입한 뒤 $N$번 자료구조에서 원소를 pop한다. 그때 pop된 원소를 $x_N$이라 한다.

• $x_N$을 리턴한다.

도현이는 길이 $M$의 수열 $C$를 가져와서 수열의 원소를 앞에서부터 차례대로 queuestack에 삽입할 것이다. 이전에 삽입한 결과는남아 있다. (예제 $1$ 참고)

queuestack에 넣을 원소들이 주어졌을 때, 해당 원소를 넣은 리턴값을 출력하는 프로그램을 작성해보자.

---

📁입력

첫째 줄에 queuestack을 구성하는 자료구조의 개수 $N$이 주어진다. ($1 \leq N \leq 100\,000$)

둘째 줄에 길이 $N$의 수열 $A$가 주어진다. $i$번 자료구조가 큐라면 $A_i = 0$, 스택이라면 $A_i = 1$이다.

셋째 줄에 길이 $N$의 수열 $B$가 주어진다. $B_i$는 $i$번 자료구조에 들어 있는 원소이다. ($1 \leq B_i \leq 1\,000\,000\,000$)

넷째 줄에 삽입할 수열의 길이 $M$이 주어진다. ($1 \leq M \leq 100\,000$)

다섯째 줄에 queuestack에 삽입할 원소를 담고 있는 길이 $M$의 수열 $C$가 주어진다. ($1 \leq C_i \leq 1\,000\,000\,000$)

입력으로 주어지는 모든 수는 정수이다.

---

📁출력

수열 $C$의 원소를 차례대로 queuestack에 삽입했을 때의 리턴값을 공백으로 구분하여 출력한다.

---

📁입출력 예

입력 1

4
0 1 1 0
1 2 3 4
3
2 4 7

출력 1

4 1 2

입력 2

5
1 1 1 1 1
1 2 3 4 5
3
1 3 5

출력 2

1 3 5

---

📁풀이

import sys
from collections import deque

n = int(sys.stdin.readline().strip())
list_a = list(map(int, sys.stdin.readline().strip().split(' ')))
list_b = list(map(int, sys.stdin.readline().strip().split(' ')))

m = int(sys.stdin.readline().strip())
list_c = list(map(int, sys.stdin.readline().strip().split(' ')))

queue = deque()

for i in range(n):
    if list_a[i] == 0:
        queue.appendleft(list_b[i])

for i in list_c:
    queue.append(i)
    print(queue.popleft(), end=" ")  

이번 문제는 사실상 큐 구현 문제다.
큐를 구현할 때 덱 함수를 불러오는 습관을 들여야겠다. from collections import deque 메모..