백준 - 2667 단지번호붙이기

문제

<그림 1>과 같이 정사각형 모양의 지도가 있다. 1은 집이 있는 곳을, 0은 집이 없는 곳을 나타낸다. 철수는 이 지도를 가지고 연결된 집의 모임인 단지를 정의하고, 단지에 번호를 붙이려 한다. 여기서 연결되었다는 것은 어떤 집이 좌우, 혹은 아래위로 다른 집이 있는 경우를 말한다. 대각선상에 집이 있는 경우는 연결된 것이 아니다. <그림 2>는 <그림 1>을 단지별로 번호를 붙인 것이다. 지도를 입력하여 단지수를 출력하고, 각 단지에 속하는 집의 수를 오름차순으로 정렬하여 출력하는 프로그램을 작성하시오.

입력

• 첫째 줄
  • 지도의 크기 N
• N개의 줄
  • 지도 데이터

제한 조건

• 5 ≤ N ≤ 25

예시

// NOTE: 입력
7
0110100
0110101
1110101
0000111
0100000
0111110
0111000

// NOTE: 출력
3
7
8
9

해결

문제의 핵심

좌표의 유효성 확인하기

solution 함수 막줄에
console.log(_result.forEach(v => console.log(v)));
라고 써서 자꾸 틀렸다.

얼탱없는 실수 🥹

bfs

• 필요한 파라미터
  • x와 y 좌표
• queue에 현재 좌표 넣어주고, 현재 좌표 visited 처리
• 반복문
  • queue에 좌표가 없을 때까지 반복
  • 상하좌우 좌표의 유효성 확인
    • 0 ≤ x ≤ n - 1
    • 0 ≤ y ≤ n - 1
    • 지도에서 1이어야 함
    • 방문하지 않은 좌표여야 함
  • 유효한 경우
    • visit 처리
    • 세대수 저장
    • 현재 좌표 queue에 push
• 반복문을 마치면 세대수를 출력

const bfs = (x, y) => {
    const _queue = [[x, y]];
    let _count = 1;
    _visited[y][x] = true;
    
    while(_queue.length) {
        const [_curX, _curY] = _queue.shift();
        
        for (let [dx, dy] of _d) {
            const [_newX, _newY] = [dx + _curX, dy + _curY];
            
            if (_newX < 0 || _newX > N - 1) continue;
            if (_newY < 0 || _newY > N - 1) continue;
            if (_graph[_newY][_newX] == 0) continue;
            if (_visited[_newY][_newX]) continue;
            
            _visited[_newY][_newX] = true;
            _count += 1;
            _queue.push([_newX, _newY]);
        }
    }
    
    return _count;
}

solution

• 완전탐색
  • 지도에서 1이며, 방문하지 않은 경우에만 dfs
• 세대수를 오름차순으로 console

const solution = () => {
    let _result = [];
    
    for (let i = 0; i < N; i++) {
        for (let j = 0; j < N; j++) {
            if (_graph[i][j] == 0 || _visited[i][j]) continue;
            
            _result.push(bfs(j, i));
        }
    }
    
    _result = _result.sort((a, b) => a - b);
    console.log(_result.length);
    _result.forEach(v => console.log(v));
};

전체 코드

// 전역변수
let [N,..._inputs] = require('fs')
    .readFileSync('dev/stdin')
    .toString()
    .trim()
    .split(/\n/);

N = +N;
const _d = [[-1, 0], [1, 0], [0, 1], [0, -1]];
const _graph = _inputs.map(v => v.split('').map(v => +v));
const _visited = Array.from(Array(N), () => Array(N).fill(false));

// NOTE: solution
const bfs = (x, y) => {
    const _queue = [[x, y]];
    let _count = 1;
    _visited[y][x] = true;
    
    while(_queue.length) {
        const [_curX, _curY] = _queue.shift();
        
        for (let [dx, dy] of _d) {
            const [_newX, _newY] = [dx + _curX, dy + _curY];
            
            if (_newX < 0 || _newX > N - 1) continue;
            if (_newY < 0 || _newY > N - 1) continue;
            if (_graph[_newY][_newX] == 0) continue;
            if (_visited[_newY][_newX]) continue;
            
            _visited[_newY][_newX] = true;
            _count += 1;
            _queue.push([_newX, _newY]);
        }
    }
    
    return _count;
}

const solution = () => {
    let _result = [];
    
    for (let i = 0; i < N; i++) {
        for (let j = 0; j < N; j++) {
            if (_graph[i][j] == 0 || _visited[i][j]) continue;
            
            _result.push(bfs(j, i));
        }
    }
    
    _result = _result.sort((a, b) => a - b);
    console.log(_result.length);
    _result.forEach(v => console.log(v));
};

solution();

---

출처

백준 2667번: 단지번호붙이기