문제 바로가기
문제:
한가롭게 방학에 놀고 있던 도현이는 갑자기 재밌는 자료구조를 생각해냈다. 그 자료구조의 이름은 queuestack이다. queuestack의 구조는 다음과 같다. 1번, 2번, ... , N번의 자료구조(queue 혹은 stack)가 나열되어있으며, 각각의 자료구조에는 한 개의 원소가 들어있다. queuestack의 작동은 다음과 같다. 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에 삽입했을 때의 리턴값을 공백으로 구분하여 출력한다.
풀이:
import sys
sys.stdin = open('/Users/song/Desktop/Python/Python/h.txt', 'r')
from collections import deque
n = int(input())
A = list(map(int,sys.stdin.readline().split(" ")))
B = list(map(int,sys.stdin.readline().split(" ")))
M = int(input())
C = list(map(int,sys.stdin.readline().split(" ")))
res = deque()
for x in range(n):
if A[x] == 0:
res.appendleft(B[x])
for i in range(M):
res.append(C[i])
print(res.popleft(), end=' ')
0 이면 Que이므로 왼쪽에다가 넣어준다.
스택인 경우는 x를 삽입후에 pop하게 되어도 변화가 없다. 무시한다.
que에 넣은 값은 살려두고, 나머지 길이 는 C배열에서 채우면 된다.
댓글