백준 24511번 : queuestack

문제

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

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

queuestack의 작동은 다음과 같다.

•  �0$x_0$을 입력받는다.
•  �0$x_0$을 1$1$번 자료구조에 삽입한 뒤 1$1$번 자료구조에서 원소를 pop한다. 그때 pop된 원소를 �1$x_1$이라 한다.
•  �1$x_1$을 2$2$번 자료구조에 삽입한 뒤 2$2$번 자료구조에서 원소를 pop한다. 그때 pop된 원소를 �2$x_2$이라 한다.
• ...
•  ��−1$x_{N-1}$을 �$N$번 자료구조에 삽입한 뒤 �$N$번 자료구조에서 원소를 pop한다. 그때 pop된 원소를 ��$x_N$이라 한다.
•  ��$x_N$을 리턴한다.

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

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

입력

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

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

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

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

다섯째 줄에 queuestack에 삽입할 원소를 담고 있는 길이 �$M$의 수열 �$C$가 주어진다. (1≤��≤1000000000$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

각 상태에 대한 큐스택 내부를 표현하면 다음과 같다.

• 초기 상태 : [1,2,3,4]$[1, 2, 3, 4]$ 
• 첫 번째 원소 삽입 : [2,2,3,1]$[2, 2, 3, 1]$ 
• 두 번째 원소 삽입 : [4,2,3,2]$[4, 2, 3, 2]$ 
• 세 번째 원소 삽입 : [7,2,3,4]$[7, 2, 3, 4]$ 

 

풀이

 

코드(시간초과 발생)

import sys
from collections import deque

def queuestack(N, A, B, M, C):
    data_structures = [deque([B[i]]) for i in range(N)]  # N개의 자료구조를 초기화

    result = []  # 결과를 저장할 리스트

    for value in C:
        for i in range(N):
            if A[i] == 0:  # 큐인 경우
                data_structures[i].append(value)
                result.append(data_structures[i].popleft())
            else:  # 스택인 경우
                data_structures[i].append(value)
                result.append(data_structures[i].pop())

    return result

N = int(input())
A = list(map(int, input().split()))
B = list(map(int, input().split()))
M = int(input())
C = list(map(int, input().split()))

result = queuestack(N, A, B, M, C)

print(*result)

 

 

• 스택에 원소를 새로 넣었다 빼는 것은 무의미하므로 무시하고 큐만 고려하면 된다.
  • 큐에 해당하는 원소들을 하나의 큐로 만든다.

import sys
from collections import deque

n = int(sys.stdin.readline())
list_a = list(map(int, sys.stdin.readline().split()))  # 0 1 1 0 (0 = queue, 1 = stack)
list_b = list(map(int, sys.stdin.readline().split()))  # 1 2 3 4

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

res = deque()

for qs in range(n):
    if list_a[qs] == 0:
        res.appendleft(list_b[qs])
        
for i in range(m):
    res.append(list_c[i])
    print(res.popleft(), end=' ')