[javascript] 백준 10026번 적록색약

문제보기

const fs = require("fs");
const filePath = process.platform === "linux" ? "/dev/stdin" : "./input.txt";
let input = fs.readFileSync(filePath).toString().trim().split("\n");

const N = Number(input.shift());
const map = input.map((item) => item.split(""));
const direction = [
    [-1, 0],
    [1, 0],
    [0, -1],
    [0, 1],
];
const BFS = (start, map, N, weakness, visited, color) => { // 각 시작좌표, 맵, N, 색약여부, 방문여부, 시작좌표의 색
    let queue = [start];
    let count = 0;
    while (queue.length) {
        let [x, y] = queue.shift();
        if (visited[x][y]) continue;
        visited[x][y] = true;
        count++;
        for (let i = 0; i < direction.length; i++) {
            let [dx, dy] = [x + direction[i][0], y + direction[i][1]];
            if (dx < 0 || dy < 0 || dx >= N || dy >= N) continue;
            if (visited[dx][dy]) continue;
            if (weakness) { // 색약일 경우
                if ( // R, G는 같은색으로 취급한다.
                    (color === "R" || color === "G") &&
                    (map[dx][dy] === "R" || map[dx][dy] === "G")
                ) {
                    queue.push([dx, dy]);
                }
                if (color === "B" && map[dx][dy] === "B") {
                    queue.push([dx, dy]);
                }
            } else { // 색약이 아니면 3개의 색을 구분한다
                if (color === "R" && map[dx][dy] === "R") {
                    queue.push([dx, dy]);
                }
                if (color === "G" && map[dx][dy] === "G") {
                    queue.push([dx, dy]);
                }
                if (color === "B" && map[dx][dy] === "B") {
                    queue.push([dx, dy]);
                }
            }
        }
    }
    return count;
};

let result = [0, 0];
for (let total = 0; total < 2; total++) {
    let mode = [false, true]; // 색약여부 
    let count = 0;
    let visited = Array.from({ length: N }, () => Array(N).fill(false));
    for (let i = 0; i < N; i++) {
        for (let j = 0; j < N; j++) {
            if (visited[i][j]) continue;
            count = BFS([i, j], map, N, mode[total], visited, map[i][j]);
            if (count > 0) result[total] += 1; // BFS에서 반환된 값이 0보다 크면 값을 1 올려준다
        }
    }
}

console.log(result.join(" "));