알고리즘/BFS & DFS

[백준/BFS&DFS/C++] 1012번 유기농 배추

데메즈 2023. 2. 28. 12:24
728x90
반응형

문제는 여기!

#include <bits/stdc++.h>

using namespace std;
int T, M, N, K;
int MAP[51][51];
bool isVisited[51][51];
int dx[] = {-1, 1, 0, 0};
int dy[] = {0, 0, -1, 1};

void bfs(int a, int b){
    queue<pair<int, int>> q;
    q.push({a,b});

    while(!q.empty()){
        int x = q.front().first;
        int y = q.front().second;
        q.pop();
        isVisited[x][y] = true;

        for(int i=0; i<4; i++){
            int nx = x + dx[i];
            int ny = y + dy[i];
            if(nx<0 || ny<0 || nx>=M || ny>=N) continue;
            if(!isVisited[nx][ny] && MAP[nx][ny]==1){
                isVisited[nx][ny] = true;
                q.push({nx, ny});
            }
        }
    }
}

void solve(){
    int count = 0;
    for(int i=0; i<M; i++){
        for(int j=0; j<N; j++){
            if(!isVisited[i][j] && MAP[i][j] == 1){
                bfs(i, j);
                count++;
            }
        }
    }
    cout << count << '\n';
}

void clearMap(){
    for(int i=0; i<M; i++){
        for(int j=0; j<N; j++){
            MAP[i][j] = 0;
            isVisited[i][j] = false;
        }
    }
}

void input() {
    cin >> T;
    for(int j=0; j<T; j++){
        cin >> M >> N >> K; //가로길이, 세로길이, 배추개수
        for(int i=0; i<K; i++){
            int x, y;
            cin >> x >> y;
            MAP[x][y] = 1;
        }
        solve();
        clearMap();
    }
}

int main() {
    ios::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);

    input();


    return 0;
}
728x90
반응형