[백준] 14716 현수막 Node.js (DFS/BFS 풀이)

문제

ANT가 처음 알고리즘 대회를 개최하게 되면서 현수막을 내걸었다.

저번 학기 영상처리 수업을 듣고 배웠던 지식을 최대한 응용 해보고 싶은 혁진이는 이 현수막에서 글자가 몇 개인지 알아보는 프로그램을 만들려 한다.

혁진이는 우선 현수막에서 글자인 부분은 1, 글자가 아닌 부분은 0으로 바꾸는 필터를 적용하여 값을 만드는데 성공했다.

그런데 혁진이는 이 값을 바탕으로 글자인 부분 1이 상, 하, 좌, 우, 대각선으로 인접하여 서로 연결되어 있다면 한 개의 글자라고 생각만 하였다.

혁진이가 필터를 적용하여 만든 값이 입력으로 주어졌을 때, 혁진이의 생각대로 프로그램을 구현하면 글자의 개수가 몇 개인지 출력하여라.

입력

첫 번째 줄에는 현수막의 크기인 M와 N가 주어진다. (1 ≤ M, N ≤ 250)

두 번째 줄부터 M+1 번째 줄까지 현수막의 정보가 1과 0으로 주어지며, 1과 0을 제외한 입력은 주어지지 않는다.

출력

혁진이의 생각대로 프로그램을 구현했을 때, 현수막에서 글자의 개수가 몇 개인지 출력하여라.

예제 입력 1

8 19
0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0
0 0 0 1 0 0 0 1 0 0 0 1 0 1 1 1 1 1 0
0 0 1 0 1 0 0 1 1 0 0 1 0 0 0 1 0 0 0
0 1 0 0 0 1 0 1 0 1 0 1 0 0 0 1 0 0 0
0 1 1 1 1 1 0 1 0 1 0 1 0 0 0 1 0 0 0
0 1 0 0 0 1 0 1 0 0 1 1 0 0 0 1 0 0 0
0 1 0 0 0 1 0 1 0 0 0 1 0 0 0 1 0 0 0
0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0

예제 출력 1

3

---

문제풀이

✅ 답안 #1 : DFS - 재귀함수로 풀이

const filePath = process.platform === 'linux' ? '/dev/stdin' : './input.txt';
const banner = require('fs')
  .readFileSync(filePath)
  .toString()
  .trim()
  .split('\n')
  .map((v) => v.split(' ').map(Number));
const [row, col] = banner.shift();

// 인접한 여덟 곳(좌,우,상,하,대각선 네 군데) 탐색할 x,y좌표
const dir = [[-1, 0], [-1, 1], [0, 1], [1, 1], [1, 0], [1, -1], [0, -1], [-1, -1]];

const dfs = (r, c) => {
  // 현재 위치 기준 인접한 8군데 탐색할 반복문
  for (let i = 0; i < 8; i++) {
    const nx = r + dir[i][0];
    const ny = c + dir[i][1];

	// 해당 위치 그래프 범위를 벗어나지 않았고 글자인 부분(값=1)인 경우
    if (nx >= 0 && nx < row && ny >= 0 && ny < col && banner[nx][ny]) {
      banner[nx][ny] = 0; // 방문 처리
      dfs(nx, ny); // 재귀
    }
  }
};

let answer = 0; // 글자 개수 카운트할 변수
for (let i = 0; i < row; i++) {
  for (let j = 0; j < col; j++) {
    if (banner[i][j]) { // 글자인 부분(값=1)인 경우(방문 처리 안 된 경우)
      banner[i][j] = 0; // 방문 처리
      answer++; // 글자 개수 증가(DFS 실행할 때마다 개수 1씩 증가)
      dfs(i, j); // DFS 실행
    }
  }
}

console.log(answer);

✅ 답안 #2 : BFS - Queue로 풀이

const filePath = process.platform === 'linux' ? '/dev/stdin' : './input.txt';
const banner = require('fs')
  .readFileSync(filePath)
  .toString()
  .trim()
  .split('\n')
  .map((v) => v.split(' ').map(Number));
const [row, col] = banner.shift();

// 인접한 여덟 곳(좌,우,상,하,대각선 네 군데) 탐색할 x,y좌표
const dir = [[-1, 0], [-1, 1], [0, 1], [1, 1], [1, 0], [1, -1], [0, -1], [-1, -1]];

const bfs = (r, c) => {
  const queue = [[r, c]];
  while (queue.length) {
    const [cx, cy] = queue.shift();
		
	// 현재 위치 기준 인접한 8군데 탐색할 반복문
    for (let i = 0; i < 8; i++) {
      const nx = cx + dir[i][0];
      const ny = cy + dir[i][1];
			
	  // 해당 위치 그래프 범위를 벗어나지 않았고 글자인 부분(값=1)인 경우
      if (nx >= 0 && nx < row && ny >= 0 && ny < col && banner[nx][ny]) {
        banner[nx][ny] = 0; // 방문 처리
        queue.push([nx, ny]); // 해당 위치 큐에 담기
      }
    }
  }
};

let answer = 0; // 글자 개수 카운트할 변수
for (let i = 0; i < row; i++) {
  for (let j = 0; j < col; j++) {
    if (banner[i][j]) { // 글자인 부분(값=1)인 경우(방문 처리 안 된 경우)
      banner[i][j] = 0; // 방문 처리
      answer++; // 글자 개수 증가(BFS 실행할 때마다 개수 1씩 증가)
      bfs(i, j); // BFS 실행
    }
  }
}

console.log(answer);