본문 바로가기
Problem Solving/Baekjoon

[백준] 24511 queuestack - Data Structure / Java

by graycode 2023. 5. 2.

 문제 링크

 

24511번: queuestack

첫째 줄에 queuestack을 구성하는 자료구조의 개수 $N$이 주어진다. ($1 \leq N \leq 100\,000$) 둘째 줄에 길이 $N$의 수열 $A$가 주어진다. $i$번 자료구조가 큐라면 $A_i = 0$, 스택이라면 $A_i = 1$이다. 셋째 줄

www.acmicpc.net

 

 풀이 과정

 

 풀이 코드

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.LinkedList;
import java.util.Queue;
import java.util.StringTokenizer;

public class Main {

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
        StringBuilder sb = new StringBuilder();
        StringTokenizer st;

        int n = Integer.parseInt(br.readLine());

        String[] info = br.readLine().split(" ");
        String[] input = br.readLine().split(" ");

        Queue<String> q = new LinkedList<>();

        while (n-- > 0) {
            if (info[n].equals("0"))
                q.offer(input[n]);
        }

        int m = Integer.parseInt(br.readLine());

        st = new StringTokenizer(br.readLine());
        while (m-- > 0) {
            q.offer(st.nextToken());
            sb.append(q.poll()).append(" ");
        }

        bw.write(sb.toString());
        bw.flush();
    }

}

댓글