11123번 양 한마리... 양 두마리... | BFS, DFS | 백준 11123 C++ 코드

 

 

이번 포스팅은 백준 11123번 양 한마리... 양 두마리...입니다.

아래 url를 클릭하시면 백준 사이트에서 문제를 볼 수 있습니다.

www.acmicpc.net/problem/11123

11123번: 양 한마리... 양 두마리...

 

 

기본 알고리즘

BFS 너비 우선 탐색

DFS 깊이 우선 탐색

 

 

 

전체 코드

 

BFS 너비 우선 탐색

 

//백준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;
    }
}

 

 

DFS 깊이 우선 탐색

 

#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;
    }
}