[Rust로 백준 하루 하나] 4-9. 바구니 뒤집기

문제 (10811번)

도현이는 바구니를 총 N개 가지고 있고, 각각의 바구니에는 1번부터 N번까지 번호가 순서대로 적혀져 있다. 바구니는 일렬로 놓여져 있고, 가장 왼쪽 바구니를 1번째 바구니, 그 다음 바구니를 2번째 바구니, ..., 가장 오른쪽 바구니를 N번째 바구니라고 부른다.

도현이는 앞으로 M번 바구니의 순서를 역순으로 만들려고 한다. 도현이는 한 번 순서를 역순으로 바꿀 때, 순서를 역순으로 만들 범위를 정하고, 그 범위에 들어있는 바구니의 순서를 역순으로 만든다.

바구니의 순서를 어떻게 바꿀지 주어졌을 때, M번 바구니의 순서를 역순으로 만든 다음, 바구니에 적혀있는 번호를 가장 왼쪽 바구니부터 출력하는 프로그램을 작성하시오.

 

입력

첫째 줄에 N (1 ≤ N ≤ 100)과 M (1 ≤ M ≤ 100)이 주어진다.

둘째 줄부터 M개의 줄에는 바구니의 순서를 역순으로 만드는 방법이 주어진다. 방법은 i j로 나타내고, 왼쪽으로부터 i번째 바구니부터 j번째 바구니의 순서를 역순으로 만든다는 뜻이다. (1 ≤ i ≤ j ≤ N)

도현이는 입력으로 주어진 순서대로 바구니의 순서를 바꾼다.

 

출력

모든 순서를 바꾼 다음에, 가장 왼쪽에 있는 바구니부터 바구니에 적혀있는 순서를 공백으로 구분해 출력한다.

---

풀이

코드

use std::io;

fn main() {
    // n, m 입력 받기
    let mut input = String::new();
    io::stdin().read_line(&mut input).unwrap();
    let mut input_iter = input
        .split_whitespace()
        .map(|i| i.trim().parse::<i32>().unwrap());
    let n: i32 = input_iter.next().unwrap();
    let m: i32 = input_iter.next().unwrap();

    let mut bucket: Vec<i32> = (1..=n).collect();

    for _ in 0..m {
        // i, j 입력 받기
        let mut line = String::new();
        io::stdin().read_line(&mut line).unwrap();
        let mut line_iter = line
            .split_whitespace()
            .map(|i| i.trim().parse::<usize>().unwrap());
        let i: usize = line_iter.next().unwrap();
        let j: usize = line_iter.next().unwrap();

        // i ~ j 번째 바구니 뒤집기
        bucket = reverse_bucket(&mut bucket, i-1, j-1);
    }

    for item in bucket {
        print!("{} ", item);
    }
}

fn reverse_bucket(bucket: &mut Vec<i32>,
    start_index: usize,
    end_index: usize
) -> Vec<i32> {
    let head_bucket: Vec<i32> = bucket[0..start_index].to_vec();
    let mid_bucket: Vec<i32> = bucket[start_index..=end_index]
        .to_vec()
        .clone()
        .into_iter()
        .rev()
        .collect();
    let tail_bucket: Vec<i32> = bucket[end_index+1..].to_vec();
    let result = [head_bucket, mid_bucket, tail_bucket].concat();
    result
    // println!("{:?}", result);
}

해설

to_vec()

iter() vs. iter_mut() vs. into_iter()

concat()

---

추가 학습

• 특이사항 없음