룰루랄라 코딩기록장

[Baekjoon] 백준 4963번 섬의개수 본문

Algorithm/그래프

[Baekjoon] 백준 4963번 섬의개수

Jeonnnng 2019. 5. 14. 16:09

4963번 섬의 개수 문제 풀이

문제

  • 정사각형으로 이루어져 있는 섬과 바다 지도가 주어진다. 섬의 개수를 세는 프로그램을 작성하시오.
    한 정사각형과 가로, 세로 또는 대각선으로 연결되어 있는 사각형은 걸어갈 수 있는 사각형이다.

입력

  • 입력은 여러 개의 테스트 케이스로 이루어져 있다. 각 테스트 케이스의 첫째 줄에는 지도의 너비 w와 높이 h가 주어진다. w와 h는 50보다 작거나 같은 양의 정수이다.
    둘째 줄부터 h개 줄에는 지도가 주어진다. 1은 땅, 0은 바다이다.
    입력의 마지막 줄에는 0이 두 개 주어진다.

출력

  • 각 테스트 케이스에 대해서, 섬의 개수를 출력한다.

예제

입력 
1 1
0
2 2
0 1
1 0
3 2
1 1 1
1 1 1
5 4
1 0 1 0 0
1 0 0 0 0
1 0 1 0 1
1 0 0 1 0
5 4
1 1 1 0 1
1 0 1 0 1
1 0 1 0 1
1 0 1 1 1
5 5
1 0 1 0 1
0 0 0 0 0
1 0 1 0 1
0 0 0 0 0
1 0 1 0 1
0 0
출력
0
1
1
3
1
9

해결방법

  • 단지번호붙이기와 같은 풀이 방식이다.
    주의할점은 이 문제에서는 이동할 수 있는 범위가 대각선도 포함되니 때문에 x,y 좌표 설정시 대각선 이동도 고려하여 설정해야한다.
    int dx[] = { 0,0,1,-1,1,-1,1,-1 };
    int dy[] = { 1,-1,0,0 ,-1,1 ,1,-1};

    코드

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;

int maze[101][101];
int group[101][101];
int dx[] = { 0,0,1,-1,1,-1,1,-1 };
int dy[] = { 1,-1,0,0 ,-1,1 ,1,-1};
int w, h;

void dfs(int x, int y,int cnt) {
    group[x][y] = cnt;
    for (int k = 0; k < 8; k++) {
        int nx = x + dx[k];
        int ny = y + dy[k];
        if (nx >= 0 && nx < h && ny >= 0 && ny < w) {
            if (maze[nx][ny] == 1 && group[nx][ny] ==0) {
                dfs(nx, ny, cnt);
            }
        }
    }
}
int main() {
    ios_base::sync_with_stdio(false);
    while (true) {
        cin >> w >> h;
        if (w == 0 && h == 0) break;
        for (int i = 0; i < h; i++) {
            for (int j = 0; j < w; j++) {
                cin >> maze[i][j];
                group[i][j] = 0;
            }
        }
        int cnt = 0;
        for (int i = 0; i < h; i++) {
            for (int j = 0; j < w; j++) {
                if (maze[i][j] == 1 && group[i][j] == 0) {
                    dfs(i, j, ++cnt);
                }
            }
        }
        cout << cnt << "\n";
    }
    return 0;
}
Comments