[백준 알고리즘]/[C++]
11123번 양 한마리... 양 두마리... | BFS, DFS | 백준 11123 C++ 코드
말하는펭귄
2021. 2. 25. 16:03
728x90
반응형
이번 포스팅은 백준 11123번 양 한마리... 양 두마리...입니다.
아래 url를 클릭하시면 백준 사이트에서 문제를 볼 수 있습니다.
11123번: 양 한마리... 양 두마리...
얼마전에 나는 불면증에 시달렸지... 천장이 뚫어져라 뜬 눈으로 밤을 지새우곤 했었지. 그러던 어느 날 내 친구 광민이에게 나의 불면증에 대해 말했더니 이렇게 말하더군. "양이라도 세봐!"
www.acmicpc.net
기본 알고리즘
BFS 너비 우선 탐색
DFS 깊이 우선 탐색
전체 코드
BFS 너비 우선 탐색
|
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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
|
//백준11123 양 한마리... 양 두마리...
#include <iostream>
#include <queue>
using namespace std;
int T, H, W;
const int MAX = 101;
int map[MAX][MAX];
bool visited[MAX][MAX];
int dy[] = { 0,0,-1,1 };
int dx[] = { -1,1,0,0 };
queue<pair<int, int>> q;
int ans[MAX];
int cnt = 0;
void BFS(int y, int x) {
visited[y][x] = true;
q.push(make_pair(y, x));
while (!q.empty()) {
y = q.front().first;
x = q.front().second;
q.pop();
for (int i = 0; i < 4; i++) {
int ny = y + dy[i];
int nx = x + dx[i];
if (ny < 0 || nx < 0 || ny >= H || nx >= W)
continue;
if (map[ny][nx] == 1 && visited[ny][nx] == 0) {
visited[ny][nx] = true;
q.push(make_pair(ny, nx));
}
}
}
}
void reset() {
for (int i = 0; i < MAX; i++) {
for (int j = 0; j < MAX; j++) {
map[i][j] = 0;
visited[i][j] = 0;
}
}
cnt = 0;
}
int main() {
cin >> T;
for (int t = 0; t < T; t++) {
/*입력*/
cin >> H >> W;
char input[MAX][MAX];
for (int i = 0; i < H; i++) {
for (int j = 0; j < W; j++) {
cin >> input[i][j];
}
}
/*입력 치환*/
for (int i = 0; i < H; i++) {
for (int j = 0; j < W; j++) {
if (input[i][j] == '.') map[i][j] = 0;
if (input[i][j] == '#') map[i][j] = 1;
}
}
/*BFS 탐색 후 출력 배열에 저장*/
for (int i = 0; i < H; i++) {
for (int j = 0; j < W; j++) {
if (map[i][j] == 1 && visited[i][j] == 0) {
BFS(i, j);
cnt++;
}
}
}
ans[t] = cnt;
/*초기화*/
reset();
}
/*출력*/
for (int i = 0; i < T; i++) {
cout << ans[i] << endl;
}
}
|
cs |
DFS 깊이 우선 탐색
|
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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
|
#include <iostream>
using namespace std;
int T, H, W;
const int MAX = 101;
int map[MAX][MAX];
bool visited[MAX][MAX];
int dy[] = { 0,0,-1,1 };
int dx[] = { -1,1,0,0 };
int ans[MAX];
int cnt = 0;
void DFS(int y, int x) {
visited[y][x] = true;
for (int i = 0; i < 4; i++) {
int ny = y + dy[i];
int nx = x + dx[i];
if (ny < 0 || nx < 0 || ny >= H || nx >= W)
continue;
if (map[ny][nx] == 1 && visited[ny][nx] == 0) {
visited[ny][nx] = true;
DFS(ny, nx);
}
}
}
void reset() {
for (int i = 0; i < MAX; i++) {
for (int j = 0; j < MAX; j++) {
map[i][j] = 0;
visited[i][j] = 0;
}
}
cnt = 0;
}
int main() {
cin >> T;
for (int t = 0; t < T; t++) {
/*입력*/
cin >> H >> W;
char input[MAX][MAX];
for (int i = 0; i < H; i++) {
for (int j = 0; j < W; j++) {
cin >> input[i][j];
}
}
/*입력 치환*/
for (int i = 0; i < H; i++) {
for (int j = 0; j < W; j++) {
if (input[i][j] == '.') map[i][j] = 0;
if (input[i][j] == '#') map[i][j] = 1;
}
}
/*DFS 탐색 후 출력 배열에 저장*/
for (int i = 0; i < H; i++) {
for (int j = 0; j < W; j++) {
if (map[i][j] == 1 && visited[i][j] == 0) {
DFS(i, j);
cnt++;
}
}
}
ans[t] = cnt;
/*초기화*/
reset();
}
/*출력*/
for (int i = 0; i < T; i++) {
cout << ans[i] << endl;
}
}
|
cs |
728x90
반응형