[백준] 3187 양치기 꿍 Node.js (DFS 풀이)

Janet·2023년 5월 22일
0

Algorithm

목록 보기
202/314

문제

양치기 꿍은 맨날 늑대가 나타났다고 마을 사람들을 속였지만 이젠 더이상 마을 사람들이 속지 않는다. 화가 난 꿍은 복수심에 불타 아예 늑대들을 양들이 있는 울타리안에 마구 집어넣어 양들을 잡아먹게 했다.

하지만 양들은 보통 양들이 아니다. 같은 울타리 영역 안의 양들의 숫자가 늑대의 숫자보다 더 많을 경우 늑대가 전부 잡아먹힌다. 물론 그 외의 경우는 양이 전부 잡아먹히겠지만 말이다.

꿍은 워낙 똑똑했기 때문에 이들의 결과는 이미 알고있다. 만약 빈 공간을 '.'(점)으로 나타내고 울타리를 '#', 늑대를 'v', 양을 'k'라고 나타낸다면 여러분은 몇 마리의 양과 늑대가 살아남을지 계산할 수 있겠는가?

단, 울타리로 막히지 않은 영역에는 양과 늑대가 없으며 양과 늑대는 대각선으로 이동할 수 없다.

입력

입력의 첫 번째 줄에는 각각 영역의 세로와 가로의 길이를 나타내는 두 개의 정수 R, C (3 ≤ R, C ≤ 250)가 주어진다.

다음 각 R줄에는 C개의 문자가 주어지며 이들은 위에서 설명한 기호들이다.

출력

살아남게 되는 양과 늑대의 수를 각각 순서대로 출력한다.

예제 입력 1

6 6
...#..
.##v#.
#v.#.#
#.k#.#
.###.#
...###

예제 출력 1

0 2

예제 입력 2

8 8
.######.
#..k...#
#.####.#
#.#v.#.#
#.#.k#k#
#k.##..#
#.v..v.#
.######.

예제 출력 2

3 1

예제 입력 3

9 12
.###.#####..
#.kk#...#v#.
#..k#.#.#.#.
#..##k#...#.
#.#v#k###.#.
#..#v#....#.
#...v#v####.
.####.#vv.k#
.......####.

예제 출력 3

3 5

문제풀이

✅ 답안 : 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 field = input.map((v) => v.split(''));
const dir = [[-1, 0], [1, 0], [0, 1], [0, -1]]; // 현재 위치 기준 인접한 네 곳 x,y좌표
let [sheep, wolf] = [0, 0]; // [각 영역 내부의 양의 수, 각 영역 내부의 늑대의 수]

const dfs = (r, c) => {
  if (field[r][c] == 'v') wolf++;
  else if (field[r][c] == 'k') sheep++;
  field[r][c] = '#'; // 방문 처리

  // 현재 위치 기준 인접한 네방향 탐색할 반복문
  for (let i = 0; i < 4; i++) {
    const nx = r + dir[i][0];
    const ny = c + dir[i][1];
	// 해당 위치가 필드 밖으로 벗어나지 않고 울타리(#)가 아닌 경우
    if (nx >= 0 && nx < R && ny >= 0 && ny < C && field[nx][ny] != '#') {
      dfs(nx, ny); // 재귀
    }
  }
};

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

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

0개의 댓글