[Rust로 백준 하루 하나] 16-11. queuestack

문제 (24511번)

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

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

 

queuestack의 작동은 다음과 같다.

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

 

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

 

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

 

입력

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

 

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

 

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

 

넷째 줄에 삽입할 수열의 길이 M이 주어진다. (1≤M≤100000)

 

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

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

 

출력

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

---

풀이

코드

use std::io::{self, BufReader, BufRead, BufWriter, Write};
use std::collections::VecDeque;

fn main() {
    let mut reader = BufReader::new(io::stdin().lock());
    let mut writer = BufWriter::new(io::stdout().lock());

    let mut input = String::new();
    reader.read_line(&mut input).unwrap();
    let n: usize = input.trim().parse().unwrap();

    input.clear();
    reader.read_line(&mut input).unwrap();
    let a: Vec<i32> = input
        .trim()
        .split_whitespace()
        .map(|i| i.parse().unwrap())
        .collect();

    input.clear();
    reader.read_line(&mut input).unwrap();
    let b: Vec<i32> = input
        .trim()
        .split_whitespace()
        .map(|i| i.parse().unwrap())
        .collect();

    input.clear();
    reader.read_line(&mut input).unwrap();
    let m: usize = input.trim().parse().unwrap();

    input.clear();
    reader.read_line(&mut input).unwrap();
    let c: Vec<i32> = input
        .trim()
        .split_whitespace()
        .map(|i| i.parse().unwrap())
        .collect();

    let mut queuestack: VecDeque<i32> = VecDeque::new();
    for (i, v) in b.iter().enumerate() {
        if a[i] == 0 { queuestack.push_back(b[i]); }
    }
    // println!("{:?}", queuestack);

    for i in c {
        queuestack.push_front(i);
        let output = queuestack.pop_back().unwrap();
        write!(writer, "{output} ").unwrap();
    }
}

 

해설

use std::io::{self, BufReader, BufRead, BufWriter, Write};

fn main() {
    let mut reader = BufReader::new(io::stdin().lock());
    let mut writer = BufWriter::new(io::stdout().lock());

    let mut n = String::new();
    reader.read_line(&mut n).unwrap();
    let n: usize = n.trim().parse().unwrap();

    let mut a = String::new();
    reader.read_line(&mut a).unwrap();
    let a: Vec<i32> = a
        .trim()
        .split_whitespace()
        .map(|i| i.parse().unwrap())
        .collect();

    let mut b = String::new();
    reader.read_line(&mut b).unwrap();
    let mut b: Vec<i32> = b
        .trim()
        .split_whitespace()
        .map(|i| i.parse().unwrap())
        .collect();

    let mut m = String::new();
    reader.read_line(&mut m).unwrap();
    let m: usize = m.trim().parse().unwrap();

    let mut c = String::new();
    reader.read_line(&mut c).unwrap();
    let c: Vec<i32> = c
        .trim()
        .split_whitespace()
        .map(|i| i.parse().unwrap())
        .collect();

    for i in 0..m {
        let mut output = c[i];
        for j in 0..n {
            if a[j] == 0 {
                let temp = output;
                output = b[j];
                b[j] = temp;
            }
        }
        write!(writer, "{output} ").unwrap();
    }
}

 

---

추가 학습

• 특이사항 없음