알고리즘
백준 4963 섬의 개수
중국고대사
2019. 4. 7. 20:42
앞에 단지번호붙이기보다 더 쉬운 문제다.
방향만 8방향으로 바뀌었을 뿐 섬과 바다만 구분하면 되기 때문에 더 쉽다.
아래는 코드..
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 | #include <iostream> #include <vector> #include <algorithm> #include <cstring> #include <queue> using namespace std; int h, w; int ans; int ar[50][50]; bool visit[50][50]; int dr[8] = { -1,-1,0,1,1, 1, 0,-1 }; int dc[8] = { 0, 1,1,1,0,-1,-1,-1 }; void dfs(int row, int col) { visit[row][col] = true; //8방향으로 check를 한다. for (int i = 0; i < 8; i++) { int nr = row + dr[i]; int nc = col + dc[i]; //배열 안에 들어오는지 검사 if (nr >= 0 && nr < h && nc >= 0 && nc < w) { //방문한 곳인지 검사 if (!visit[nr][nc] && ar[nr][nc] == 1) { dfs(nr, nc); } } } } int main(int argc, char** argv) { int test = 0; while (1) { int cnt = 0; test++; cin >> w >> h; if (h == 0 && w == 0)break; //테스트 케이스가 여러 번이므로 초기화 해준다. memset(ar, 0, sizeof(ar)); memset(visit, false, sizeof(visit)); for (int i = 0; i < h; i++) { for (int j = 0; j < w; j++) { scanf("%d", &ar[i][j]); } } for (int i = 0; i < h; i++) { for (int j = 0; j < w; j++) { //방문하지 않은 곳이면 dfs 들어간다. if (!visit[i][j] && ar[i][j] == 1) { cnt++; dfs(i, j); } } } printf("%d\n", cnt); } return 0; } | cs |