-
Notifications
You must be signed in to change notification settings - Fork 1
/
4963_DFS.cpp
63 lines (53 loc) · 1.57 KB
/
4963_DFS.cpp
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
#include <iostream>
#include <vector>
using namespace std;
const int VISITED = 2;
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 sr, int sc, int h, int w, vector<vector<int>> &map) { // dfs 탐색
map[sr][sc] = VISITED; // 방문 check
for(int i = 0; i < 8; i++) { // 이어진 땅 없는지 탐색
int nr = sr + dr[i];
int nc = sc + dc[i];
if(nr >= 0 && nr < h && nc >= 0 && nc < w && map[nr][nc] == 1) { // 땅 발견
dfs(nr, nc, h, w, map);
}
}
}
int cntIsland(int h, int w, vector<vector<int>> &map) { // 섬 개수 세기
int cnt = 0;
for(int i = 0; i < h; i++) {
for(int j = 0; j < w; j++) {
if(map[i][j] == 1) { // 땅 발견
cnt++;
dfs(i, j, h, w, map); // (i,j)와 연결된 땅 탐색 (== 섬 만들기)
}
}
}
return cnt;
}
/*
* [섬의 개수 세기]
* 1. 땅을 발견할 때마다 그래프 탐색을 이용하여 연결된 땅을 찾아 섬을 만든다.
* 2. 따라서 그래프 탐색 횟수가 섬의 개수가 된다.
*/
int main() {
int w, h;
vector<vector<int>> map;
while(true) {
// 입력
cin >> w >> h;
if(w == 0 && h == 0) { // 종료조건
break;
}
map.assign(h, vector<int> (w, 0));
for(int i = 0; i < h; i++) {
for(int j = 0; j < w; j++) {
cin >> map[i][j];
}
}
// 연산 & 출력
cout << cntIsland(h, w, map) << '\n';
}
return 0;
}