[백준 1018 / JAVA] 체스판 다시 칠하기

https://www.acmicpc.net/problem/1018

1018번: 체스판 다시 칠하기

 

    public static void main(String[] args) throws IOException {
        String input;
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        input = br.readLine();
        StringTokenizer st = new StringTokenizer(input);

        N = Integer.parseInt(st.nextToken());
        M = Integer.parseInt(st.nextToken());

        map = new char[N][M];

        for (int i = 0; i < N; i++) {
            map[i] = br.readLine().toCharArray();
        }

        for (int i = 0; i <= N - 8; i++) {
            for (int j = 0; j <= M - 8; j++) {
                //첫번쨰 B, W 중 더 적은 경우
                min = Math.min(black(i, j), white(i, j));
                if (result > min)
                    result = min;
            }
        }
        System.out.println(result);
    }

체스판을 입력하고 모든 8X8의 경우를 확인한다.

확인할 때 첫번째 칸을 W로 하는 경우, B로 하는 경우 2가지로 나뉘어 경우를 계산한다.

 

    public static int black(int a, int b) {
        int cnt = 0;
        for (int i = a; i < a + 8; i++) {
            for (int j = b; j < b + 8; j++) {
                if (i % 2 == 0) { // 홀수 줄
                    if (j % 2 == 0) {
                        if (map[i][j] != 'B')
                            cnt++;
                    } else {
                        if (map[i][j] != 'W')
                            cnt++;
                    }
                } else { // 짝수 줄
                    if (j % 2 == 0) {
                        if (map[i][j] != 'W')
                            cnt++;
                    } else {
                        if (map[i][j] != 'B')
                            cnt++;
                    }
                }
            }
        }
        return cnt;
    }

첫번째 칸을 B로 맞추는 경우이다. 홀수 줄, 짝수 줄로 나누고 그 줄에서 홀수 칸 짝수 칸으로 나누어 확인한다.

바꿔야 하는 경우의 수를 구한다.

 

    public static int white(int a, int b) {
        int cnt = 0;
        for (int i = a; i < a + 8; i++) {
            for (int j = b; j < b + 8; j++) {
                if (i % 2 == 0) { // 홀수 줄
                    if (j % 2 == 0) {
                        if (map[i][j] != 'W')
                            cnt++;
                    } else {
                        if (map[i][j] != 'B')
                            cnt++;
                    }
                } else { // 짝수 줄
                    if (j % 2 == 0) {
                        if (map[i][j] != 'B')
                            cnt++;
                    } else {
                        if (map[i][j] != 'W')
                            cnt++;
                    }
                }
            }
        }
        return cnt;
    }

첫번째 칸을 W로 맞추는 경우이다. B일 경우와 칸만 반대이다.

 

전체코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.StringTokenizer;

public class Main {
    static int N, M, min, result = Integer.MAX_VALUE;
    static char[][] map;

    public static void main(String[] args) throws IOException {
        String input;
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        input = br.readLine();
        StringTokenizer st = new StringTokenizer(input);

        N = Integer.parseInt(st.nextToken());
        M = Integer.parseInt(st.nextToken());

        map = new char[N][M];

        for (int i = 0; i < N; i++) {
            map[i] = br.readLine().toCharArray();
        }

        for (int i = 0; i <= N - 8; i++) {
            for (int j = 0; j <= M - 8; j++) {
                //첫번쨰 B, W 중 더 적은 경우
                min = Math.min(black(i, j), white(i, j));
                if (result > min)
                    result = min;
            }
        }
        System.out.println(result);
    }

    public static int black(int a, int b) {
        int cnt = 0;
        for (int i = a; i < a + 8; i++) {
            for (int j = b; j < b + 8; j++) {
                if (i % 2 == 0) { // 홀수 줄
                    if (j % 2 == 0) {
                        if (map[i][j] != 'B')
                            cnt++;
                    } else {
                        if (map[i][j] != 'W')
                            cnt++;
                    }
                } else { // 짝수 줄
                    if (j % 2 == 0) {
                        if (map[i][j] != 'W')
                            cnt++;
                    } else {
                        if (map[i][j] != 'B')
                            cnt++;
                    }
                }
            }
        }
        return cnt;
    }

    public static int white(int a, int b) {
        int cnt = 0;
        for (int i = a; i < a + 8; i++) {
            for (int j = b; j < b + 8; j++) {
                if (i % 2 == 0) { // 홀수 줄
                    if (j % 2 == 0) {
                        if (map[i][j] != 'W')
                            cnt++;
                    } else {
                        if (map[i][j] != 'B')
                            cnt++;
                    }
                } else { // 짝수 줄
                    if (j % 2 == 0) {
                        if (map[i][j] != 'B')
                            cnt++;
                    } else {
                        if (map[i][j] != 'W')
                            cnt++;
                    }
                }
            }
        }
        return cnt;
    }
}