본문 바로가기

컴퓨터 과학/자료구조, 알고리즘

[Rust로 백준 하루 하나] 12-4. 체스판 다시 칠하기

문제 (1018번)

666은 종말을 나타내는 수라고 한다. 따라서, 많은 블록버스터 영화에서는 666이 들어간 제목을 많이 사용한다. 영화감독 숌은 세상의 종말 이라는 시리즈 영화의 감독이다. 조지 루카스는 스타워즈를 만들 때, 스타워즈 1, 스타워즈 2, 스타워즈 3, 스타워즈 4, 스타워즈 5, 스타워즈 6과 같이 이름을 지었고, 피터 잭슨은 반지의 제왕을 만들 때, 반지의 제왕 1, 반지의 제왕 2, 반지의 제왕 3과 같이 영화 제목을 지었다. 하지만 숌은 자신이 조지 루카스와 피터 잭슨을 뛰어넘는다는 것을 보여주기 위해서 영화 제목을 좀 다르게 만들기로 했다.

 

종말의 수란 어떤 수에 6이 적어도 3개 이상 연속으로 들어가는 수를 말한다. 제일 작은 종말의 수는 666이고, 그 다음으로 큰 수는 1666, 2666, 3666, .... 이다. 따라서, 숌은 첫 번째 영화의 제목은 "세상의 종말 666", 두 번째 영화의 제목은 "세상의 종말 1666"와 같이 이름을 지을 것이다. 일반화해서 생각하면, N번째 영화의 제목은 세상의 종말 (N번째로 작은 종말의 수) 와 같다.

 

숌이 만든 N번째 영화의 제목에 들어간 수를 출력하는 프로그램을 작성하시오. 숌은 이 시리즈를 항상 차례대로 만들고, 다른 영화는 만들지 않는다.

 

입력

첫째 줄에 N이 주어진다. N은 10,000보다 작거나 같은 자연수이다.

 

출력

첫째 줄에 N번째 영화의 제목에 들어간 수를 출력한다.


풀이

코드

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

fn count_repaints(board: &[Vec<char>], start_row: usize, start_col: usize, pattern: char) -> usize {
    let mut repaints = 0;
    let alt_pattern = if pattern == 'W' { 'B' } else { 'W' };

    for i in 0..8 {
        for j in 0..8 {
            let expected_color = if (i + j) % 2 == 0 { pattern } else { alt_pattern };
            if board[start_row + i][start_col + j] != expected_color {
                repaints += 1;
            }
        }
    }

    repaints
}

fn main() {
    let stdin = io::stdin();
    let mut input = stdin.lock().lines();

    let first_line = input.next().unwrap().unwrap();
    let mut dimensions = first_line.split_whitespace();
    let m: usize = dimensions.next().unwrap().parse().unwrap();
    let n: usize = dimensions.next().unwrap().parse().unwrap();

    let mut board = Vec::new();

    for _ in 0..m {
        let line = input.next().unwrap().unwrap();
        board.push(line.chars().collect());
    }

    let mut min_repaints = usize::MAX;

    for i in 0..=m - 8 {
        for j in 0..=n - 8 {
            let repaints_w = count_repaints(&board, i, j, 'W');
            let repaints_b = count_repaints(&board, i, j, 'B');
            min_repaints = min_repaints.min(repaints_w.min(repaints_b));
        }
    }

    println!("{}", min_repaints);
}

해설

특이사항 없음

오답

use std::io;

fn main() {
    let mut input = String::new();
    io::stdin().read_line(&mut input).unwrap();
    let mut input_iter = input
        .split_whitespace()
        .map(|i| i.trim().parse::<usize>().unwrap());
    let n = input_iter.next().unwrap();
    let m = input_iter.next().unwrap();

    input.clear();
    for _ in 0..n {
        io::stdin().read_line(&mut input).unwrap();
    }
    let input: Vec<&str> = input.split_whitespace().collect();
    let mut entire_board: Vec<Vec<char>> = Vec::new();
    for index in 0..input.len() {
        entire_board.push(input[index].chars().collect::<Vec<char>>());
    }

    let mut output: usize = 0;
    for x in 0..=n-8 {
        for y in 0..=m-8 {
            output = count_repaint(&entire_board);
        }
    }
    println!("{}", output);


}

fn count_repaint(board: &[Vec<char>]) -> usize {
    let mut count: usize = 0;
    let mut count_vec: Vec<usize> = Vec::new();
    for i in 0..8 {
        for j in 0..8 {
            if board[0][0] == 'W' {
                if (i+j) % 2 == 0 {
                    if board[i][j] == 'B' { count += 1 }
                } else {
                    if board[i][j] == 'W' { count += 1 }
                }
            } else {
                if (i+j) % 2 == 0 {
                    if board[i][j] == 'W' { count += 1 }
                } else {
                    if board[i][j] == 'B' { count += 1 }
                }
            }
        }
    }
    count_vec.push(count);
    let result = count_vec.into_iter().min().unwrap();
    result
}

 


추가 학습

  • 특이사항 없음