🎲 백준 10026번 적록색약

10026번

🧸 처음에 BFS안에서 경우를 나누어주려고 했는데 실패했다.. 같이 풀던 친구가 말한 방법대로 하니까 성공~~!
💡 적록색약인 경우의 배열을 따로 추가해주었다.

코드

const fs = require('fs'); 
let input = fs.readFileSync('/dev/stdin').toString().trim().split('\n');
const N = +input.shift();
input = input.map((item) => item.split(""));
const inputYes = input.map((line) =>
  line.map((item) => (item = item === "G" ? "R" : item))
);

const checked = Array.from(new Array(N), () => new Array(N).fill(0));
const checkedYes = Array.from(new Array(N), () => new Array(N).fill(0));

const dir = [
  [0, 1],
  [0, -1],
  [1, 0],
  [-1, 0],
];

const BFS = (input, i, j, checked) => {
  const queue = [[i, j]];
  while (queue.length) {
    const [pY, pX] = queue.shift();
    for (let d = 0; d < 4; d++) {
      const nY = pY + dir[d][0];
      const nX = pX + dir[d][1];
      if (nX >= 0 && nY >= 0 && nX < N && nY < N && checked[nY][nX] === 0) {
        if (input[nY][nX] === input[pY][pX]) {
          queue.push([nY, nX]);
          checked[nY][nX] = 1;
        }
      }
    }
  }
};

let no = 0;
let yes = 0;
for (let i = 0; i < N; i++) {
  for (let j = 0; j < N; j++) {
    if (checked[i][j] === 0) {
      BFS(input, i, j, checked);
      no++;
    }
  }
}

for (let i = 0; i < N; i++) {
  for (let j = 0; j < N; j++) {
    if (checkedYes[i][j] === 0) {
      BFS(inputYes, i, j, checkedYes);
      yes++;
    }
  }
}

console.log(no, yes);