백준 알고리즘

[Silver III] 칸토어 집합 - 4779

문제 링크

성능 요약

메모리: 13680 KB, 시간: 4 ms

분류

분할 정복, 재귀

문제 설명

칸토어 집합은 0과 1사이의 실수로 이루어진 집합으로, 구간 [0, 1]에서 시작해서 각 구간을 3등분하여 가운데 구간을 반복적으로 제외하는 방식으로 만든다.

전체 집합이 유한이라고 가정하고, 다음과 같은 과정을 통해서 칸토어 집합의 근사를 만들어보자.

1. -가 3^N개 있는 문자열에서 시작한다.

2. 문자열을 3등분 한 뒤, 가운데 문자열을 공백으로 바꾼다. 이렇게 하면, 선(문자열) 2개가 남는다.

3. 이제 각 선(문자열)을 3등분 하고, 가운데 문자열을 공백으로 바꾼다. 이 과정은 모든 선의 길이가 1일때 까지 계속 한다.

예를 들어, N=3인 경우, 길이가 27인 문자열로 시작한다.

---------------------------

여기서 가운데 문자열을 공백으로 바꾼다.

---------         ---------

남은 두 선의 가운데 문자열을 공백으로 바꾼다.

---   ---         ---   ---

한번 더

- -   - -         - -   - -

모든 선의 길이가 1이면 멈춘다. N이 주어졌을 때, 마지막 과정이 끝난 후 결과를 출력하는 프로그램을 작성하시오.

입력

입력을 여러 줄로 이루어져 있다. 각 줄에 N이 주어진다. 파일의 끝에서 입력을 멈춘다. N은 0보다 크거나 같고, 12보다 작거나 같은 정수이다.

출력

입력으로 주어진 N에 대해서, 해당하는 칸토어 집합의 근사를 출력한다.

Code

use std::io::{self, BufRead};

fn go(s: &mut [u8], start: usize, end: usize) {
    if end - start == 1 {
        return;
    }

    let k = (end - start) / 3;
    for i in start + k..end - k {
        s[i] = b' ';
    }

    go(s, start, start + k);
    go(s, end - k, end);
}

fn main() {
    let stdin = io::stdin();
    let mut lines = stdin.lock().lines().map(|line| line.unwrap());

    while let Some(line) = lines.next() {
        let n: usize = line.trim().parse().unwrap();
        let size = 3usize.pow(n as u32);

        let mut s = vec![b'-'; size];
        go(&mut s, 0, size);

        let output = String::from_utf8(s).unwrap();
        println!("{}", output);
    }
}

백준 4779 - 칸토어 집합 [Rust]