백준 1388: 바닥 장식 - javascript(이중 for문, dfs)

문제 출처 

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

1388번: 바닥 장식

문제

형택이는 건축가이다. 지금 막 형택이는 형택이의 남자 친구 기훈이의 집을 막 완성시켰다. 형택이는 기훈이 방의 바닥 장식을 디자인했고, 이제 몇 개의 나무 판자가 필요한지 궁금해졌다. 나무 판자는 크기 1의 너비를 가졌고, 양수의 길이를 가지고 있다. 기훈이 방은 직사각형 모양이고, 방 안에는 벽과 평행한 모양의 정사각형으로 나누어져 있다.

이제 ‘-’와 ‘|’로 이루어진 바닥 장식 모양이 주어진다. 만약 두 개의 ‘-’가 인접해 있고, 같은 행에 있다면, 두 개는 같은 나무 판자이고, 두 개의 ‘|’가 인접해 있고, 같은 열에 있다면, 두 개는 같은 나무 판자이다.

기훈이의 방 바닥을 장식하는데 필요한 나무 판자의 개수를 출력하는 프로그램을 작성하시오.

입력

첫째 줄에 방 바닥의 세로 크기N과 가로 크기 M이 주어진다. 둘째 줄부터 N개의 줄에 M개의 문자가 주어진다. 이것은 바닥 장식 모양이고, '-‘와 ’|‘로만 이루어져 있다. N과 M은 50 이하인 자연수이다.

출력

첫째 줄에 문제의 정답을 출력한다.

예제 입력 1 복사

4 4
----
----
----
----

예제 출력 1 복사

4

예제 입력 2 복사

6 9
-||--||--
--||--||-
|--||--||
||--||--|
-||--||--
--||--||-

예제 출력 2 복사

31

예제 입력 3 복사

7 8
--------
|------|
||----||
|||--|||
||----||
|------|
--------

예제 출력 3 복사

13

예제 입력 4 복사

10 10
||-||-|||-
||--||||||
-|-|||||||
-|-||-||-|
||--|-||||
||||||-||-
|-||||||||
||||||||||
||---|--||
-||-||||||

예제 출력 4 복사

41

예제 입력 5 복사

6 6
-||--|
||||||
|||-|-
-||||-
||||-|
||-||-

예제 출력 5 복사

19

 

문제 풀이 힌트 

이중 for문을 사용할 수도 있고, dfs를 활용할수도 있다. 

지난 시간에 for문을 활용한 코드를 작성했으니, 

https://fordang.tistory.com/113

백준 1388: 바닥 장식 - javascript(이중 for문, dfs)

 

이번에는 dfs를 활용해서 코드를 작성해보자. 

문제 풀이 코드 

02. DFS

const input = require('fs').readFileSync('/dev/stdin').toString().trim().split('\n'); 
const [N,M] = input.shift().split(' ').map(v => +v); 
const room = []; 
for(let i=0; i<N; i++){
    room.push(input[i].split('')); 
}
const visited = Array.from({length: N}, () => new Array(M).fill(false)); //dfs를 할 때는 해당 요소를 방문했는지 체크를 해줘야 한다. 
const d = [-1, 1]; // '-'일때는, 좌우로만 이동. '|'일때는, 위아래로 이동을 의미. 

function dfs(x, y, item){
    visited[y][x] = true; //방문한거 체크 
    if(item === '-'){ // '-'일 때는 좌우로 이동 해
        for(let i=0; i<2; i++){
        const nx = x + d[i]; //좌우로 이동 
        if (nx < 0 || nx >= M) continue; //room 그래프를 넘어가면 안됨 
        if(!visited[y][nx] && room[y][nx] === '-'){ // 방문한적 없으면서, '-'일 때
            dfs(nx, y, item)
        }
    }    
   }else{// '|'일 때 상하로 이동 
       for(let i=0; i<2; i++){
           const ny = y + d[i]; // 상하로 이동 
           if(ny < 0 || ny >= N) continue; //room 그래프를 벗어나면 안됨 
           if(!visited[ny][x] && room[ny][x] === '|'){ //방문한적 없으면서 '|'일 때
               dfs(x, ny, item)
           }
       }
   }    
}

//room 그래프 하나하나 다 돌것이다. 
let count = 0; 
for(let i=0; i<M; i++){ //가로 idx
    for(let j=0; j<N; j++){ //세로 idx
        if(!visited[j][i]){//방문하지 않았을 때만, 
           count++; //바닥 장식 하나 더 사용한 것임. 
           room[j][i] === '-' ? dfs(i,j,'-') : dfs(i,j,'|') // '-' 인지, '|'인지에 따라 item을 다르게 넣어주는 것 . 
           }
    }
}
console.log(count);

Only 풀이 코드 

02. DFS

const input = require('fs').readFileSync('/dev/stdin').toString().trim().split('\n'); 
const [N,M] = input.shift().split(' ').map(v => +v); 
const room = []; 
for(let i=0; i<N; i++){
    room.push(input[i].split('')); 
}
const visited = Array.from({length: N}, () => new Array(M).fill(false)); 
const d = [-1, 1]; 
function dfs(x, y, item){
    visited[y][x] = true; 
    if(item === '-'){
        for(let i=0; i<2; i++){
        const nx = x + d[i]; 
        if (nx < 0 || nx >= M) continue; 
        if(!visited[y][nx] && room[y][nx] === '-'){
            dfs(nx, y, item)
        }
    }    
   }else{
       for(let i=0; i<2; i++){
           const ny = y + d[i]; 
           if(ny < 0 || ny >= N) continue; 
           if(!visited[ny][x] && room[ny][x] === '|'){
               dfs(x, ny, item)
           }
       }
   }    
}
let count = 0; 
for(let i=0; i<N; i++){
    for(let j=0; j<M; j++){
        if(!visited[i][j]){
           count++; 
           room[i][j] === '-' ? dfs(j,i,'-') : dfs(j,i,'|')
           }
    }
}
console.log(count);