알고리즘/BFS & DFS

[백준/BFS&DFS/C++] 1926번 그림

데메즈 2023. 3. 2. 16:54
728x90
반응형

문제는 여기!

#include <bits/stdc++.h>

using namespace std;
int N, M;
int MAP[501][501];
int dx[] = {-1, 1, 0, 0};
int dy[] = {0, 0, -1, 1};
bool isVisited[501][501];
int printCnt = 0, maxCnt = 0;

void bfs(int left, int right){
    queue<pair<int, int>> q;
    q.push({left, right});
    isVisited[left][right] = true;
    int cnt = 1;

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

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

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

void input() {
    cin >> N >> M;
    for(int i=0; i<N; i++){
        for(int j=0; j<M; j++){
            cin >> MAP[i][j];
        }
    }
}

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

    input();
    solve();
    cout << printCnt << '\n';
    cout << maxCnt;

    return 0;
}
728x90
반응형