백준 3184: 양 - javascript(dfs)

문제 출처 

https://www.acmicpc.net/problem/3184

3184번: 양

 

문제 

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

마당은 행과 열로 이루어진 직사각형 모양이다. 글자 '.' (점)은 빈 필드를 의미하며, 글자 '#'는 울타리를, '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

문제 풀이 힌트 

 . = 빈 필드 
 '#' = 울타리 
 'o' = 양 
 'v' = 늑대 
 수평 수직 이동 
 이동할 수 있는 영역은 같은 영역 
 영역 안 양의 수가 늑대의 수보다 많으면 늑대 쫓아내 
 그 반대는 늑대 승으로, 양 쫓아내 
 출력 : 남아있는 양과 늑대 수 출력 

 

dfs 로 마당 각 요소를 방문하고, 같은 영역의 양과 늑대의 수를 센다. 

수를 비교해서 남아있는 양과 늑대의 수에 더해주고 

총 값을 출력하자. 

문제 풀이 코드 

const input = require('fs').readFileSync('/dev/stdin').toString().trim().split("\n"); 
const [R,C] = input.shift().split(' ').map(v => +v); //R은 세로줄, C는 가로줄
const graph = Array.from({length: R}).map(() => []); //마당 2차원 배열로 만들기 
for(let i=0; i<R; i++){
    graph[i] = input[i].split('') //그래프에 마당 값 채워주기 
}
const visited = Array.from({length: R}).map(() => new Array(C).fill(false)); //grpah의 각 요소 방문했는지 체크 

const dx = [-1,1,0,0] //왼쪽, 오른쪽, 아래, 위 이동 가능  
const dy = [0,0,-1,1]

let sheep = 0; //같은 영역에 있는 양의 수 
let wolf = 0; //같은 영역에 있는 늑대 수 
function dfs(y,x){
    if(graph[y][x] === 'o') sheep++; //지금 방문한 그래프 요소가 양일 때   
    if(graph[y][x] === 'v') wolf++; //지금 방문한 그래프 요소가 늑대일 때 
    for(let i=0; i<4; i++){ //방문한 요소의 왼,오,아,위 확인 후 이동 해보자. 
        let nx = x + dx[i]
        let ny = y + dy[i]
        if(nx < 0 || ny < 0 || nx>=C || ny >=R) continue; //그래프를 벗어나는 곳은 안됨
        if(!visited[ny][nx] && graph[ny][nx] !== '#' ){ //방문한적 x &  울타리가 아니어야 함
            visited[ny][nx] = true; //방문한 거 체크 
            dfs(ny, nx); //방문하자. 
        }
    }
}
let ans1 = 0; //남아있는 양 수 
let ans2 = 0; //남아있는 늑대 수 
for(let i=0; i<R; i++){ //세로줄 
    for(let j=0; j<C; j++){ //가로줄
        if(graph[i][j] !== '#' && !visited[i][j]){ //울타리가 아니고, 방문한적 x
           visited[i][j] = true; //방문 체크 
           dfs(i,j);
           sheep > wolf ? (ans1 += sheep) : (ans2 += wolf) //양 수가 늑대 수보다 많으면 ? 양 수만 더해줘. 아니라면 늑대 수만 더해줘 
           sheep = 0; //또 새로운 영역의 양과 늑대 수를 세기 위해 0으로 초기화 해야함 
           wolf = 0; 
        }
    }
}
console.log(ans1, ans2);

 

Only 문제 풀이

const input = require('fs').readFileSync('/dev/stdin').toString().trim().split("\n"); 
const [R,C] = input.shift().split(' ').map(v => +v); 
const graph = Array.from({length: R}).map(() => []); 
for(let i=0; i<R; i++){
    graph[i] = input[i].split('')
}
const visited = Array.from({length: R}).map(() => new Array(C).fill(false)); 

const dx = [-1,1,0,0]; 
const dy = [0,0,-1,1]

let sheep = 0;
let wolf = 0; 
function dfs(y,x){
    if(graph[y][x] === 'o') sheep++;  
    if(graph[y][x] === 'v') wolf++; 
    for(let i=0; i<4; i++){
        let nx = x + dx[i]
        let ny = y + dy[i]
        if(nx < 0 || ny < 0 || nx>=C || ny >=R) continue; 
        if(!visited[ny][nx] && graph[ny][nx] !== '#' ){
            visited[ny][nx] = true; 
            dfs(ny, nx); 
        }
    }
}
let ans1 = 0; 
let ans2 = 0;
for(let i=0; i<R; i++){
    for(let j=0; j<C; j++){
        if(graph[i][j] !== '#' && !visited[i][j]){
           visited[i][j] = true; 
           dfs(i,j);
           sheep > wolf ? (ans1 += sheep) : (ans2 += wolf)
           sheep = 0; 
           wolf = 0; 
        }
    }
}
console.log(ans1, ans2);