[백준] 3184 양 Node.js (DFS 풀이)

Janet·2023년 5월 17일
0

Algorithm

목록 보기
199/314

문제

미키의 뒷마당에는 특정 수의 양이 있다. 그가 푹 잠든 사이에 배고픈 늑대는 마당에 들어와 양을 공격했다.

마당은 행과 열로 이루어진 직사각형 모양이다. 글자 '.' (점)은 빈 필드를 의미하며, 글자 '#'는 울타리를, 'o'는 양, 'v'는 늑대를 의미한다.

한 칸에서 수평, 수직만으로 이동하며 울타리를 지나지 않고 다른 칸으로 이동할 수 있다면, 두 칸은 같은 영역 안에 속해 있다고 한다. 마당에서 "탈출"할 수 있는 칸은 어떤 영역에도 속하지 않는다고 간주한다.

다행히 우리의 양은 늑대에게 싸움을 걸 수 있고 영역 안의 양의 수가 늑대의 수보다 많다면 이기고, 늑대를 우리에서 쫓아낸다. 그렇지 않다면 늑대가 그 지역 안의 모든 양을 먹는다.

맨 처음 모든 양과 늑대는 마당 안 영역에 존재한다.

아침이 도달했을 때 살아남은 양과 늑대의 수를 출력하는 프로그램을 작성하라.

입력

첫 줄에는 두 정수 R과 C가 주어지며(3 ≤ R, C ≤ 250), 각 수는 마당의 행과 열의 수를 의미한다.

다음 R개의 줄은 C개의 글자를 가진다. 이들은 마당의 구조(울타리, 양, 늑대의 위치)를 의미한다.

출력

하나의 줄에 아침까지 살아있는 양과 늑대의 수를 의미하는 두 정수를 출력한다.

예제 입력 1

6 6
...#..
.##v#.
#v.#.#
#.o#.#
.###.#
...###

예제 출력 1

0 2

예제 입력 2

8 8
.######.
#..o...#
#.####.#
#.#v.#.#
#.#.o#o#
#o.##..#
#.v..v.#
.######.

예제 출력 2

3 1

예제 입력 3

9 12
.###.#####..
#.oo#...#v#.
#..o#.#.#.#.
#..##o#...#.
#.#v#o###.#.
#..#v#....#.
#...v#v####.
.####.#vv.o#
.......####.

예제 출력 3

3 5

문제풀이

💡 문제풀이 과정

  1. 수평 혹은 수직으로 울타리가 아닌 곳 이어야 이동 가능 (상,하,좌,우 인접한 네 곳 x,y좌표 탐색)
  2. 각 영역 안에 양의 수와 늑대의 수가 몇 마리인지 파악 (DFS 내부에서 양과 늑대의 수 카운팅하여 반환)
  3. 한 영역 안에 양 > 늑대인 경우, 양은 살고 해당 영역 늑대의 수 = 0
  4. 한 영역 안에 양 <= 늑대인 경우, 늑대는 살고 해당 영역의 양의 수 = 0
  5. DFS 외부에서 DFS 결과값으로 영역마다의 양과 늑대 마리 수 반환받고 합산

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

const filePath = process.platform === 'linux' ? '/dev/stdin' : './input.txt';
const input = require('fs').readFileSync(filePath).toString().trim().split('\n');
const [R, C] = input.shift().split(' ').map(Number);
const yard = input.map((v) => v.split(''));
const dir = [[-1, 0], [1, 0], [0, 1], [0, -1]]; // 현재 위치 기준 인접한 네 곳 x,y좌표

// [각 영역 내부의 양의 수, 각 영역 내부의 늑대의 수, 전체 양의 수, 전체 늑대의 수]
let [sheep, wolf, totalSheep, totalWolf] = [0, 0, 0, 0];

const dfs = (row, col) => {
  if (yard[row][col] == 'o') sheep++;
  else if (yard[row][col] == 'v') wolf++;
  yard[row][col] = '#'; // 방문 처리

  // 현재 위치 기준 인접한 네 곳 탐색할 반복문
  for (let i = 0; i < 4; i++) {
    const x = row + dir[i][0];
    const y = col + dir[i][1];
	// 해당 위치가 마당(그래프) 밖으로 벗어나지 않고 울타리(#)가 아닌 경우
    if (x >= 0 && x < R && y >= 0 && y < C && yard[x][y] != '#') {
      dfs(x, y);
    }
  }
};

for (let i = 0; i < R; i++) {
  for (let j = 0; j < C; j++) {
	// 울타리(#)가 아닌(+방문 처리 안 된) 경우
    if (yard[i][j] != '#') {
      dfs(i, j); // DFS 실행
	  // 양 > 늑대인 경우: 양 전체 마리 수에 카운트한 양의 수 더함,
	  // 양 <= 늑대인 경우: 늑대 전체 마리 수에 카운트한 늑대의 수 더함
      sheep > wolf ? (totalSheep += sheep) : (totalWolf += wolf);
      sheep = 0; // DFS 실행 후 결과값 초기화
      wolf = 0; // DFS 실행 후 결과값 초기화
    }
  }
}

console.log(totalSheep + ' ' + totalWolf);
profile
😸

0개의 댓글