[백준 - 1926] 그림 - C++

분류: BFS /

 

문제

문제 링크

문제 설명

어떤 큰 도화지에 그림이 그려져 있을 때, 그 그림의 개수와, 그 그림 중 넓이가 가장 넓은 것의 넓이를 출력하여라. 단, 그림이라는 것은 1로 연결된 것을 한 그림이라고 정의하자. 가로나 세로로 연결된 것은 연결이 된 것이고 대각선으로 연결이 된 것은 떨어진 그림이다. 그림의 넓이란 그림에 포함된 1의 개수이다.

입력

첫째 줄에 도화지의 세로 크기 n(1 ≤ n ≤ 500)과 가로 크기 m(1 ≤ m ≤ 500)이 차례로 주어진다. 두 번째 줄부터 n+1 줄 까지 그림의 정보가 주어진다. (단 그림의 정보는 0과 1이 공백을 두고 주어지며, 0은 색칠이 안된 부분, 1은 색칠이 된 부분을 의미한다)

출력

첫째 줄에는 그림의 개수, 둘째 줄에는 그 중 가장 넓은 그림의 넓이를 출력하여라. 단, 그림이 하나도 없는 경우에는 가장 넓은 그림의 넓이는 0이다.

 

풀이

#include <iostream>
#include <queue>
using namespace std;
#define X first
#define Y second

int graph[501][501];
bool vis[501][501];
int dx[4] = {1, 0 , -1, 0};
int dy[4] = {0, 1 , 0, -1};

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

    int N, M;
    cin >> N >> M;
    for(int i=0; i<N; i++)
        for(int j=0; j<M; j++)
            cin >> graph[i][j];

    int cnt=0, wide=0;
    queue<pair<int,int>> Q;
    for(int i=0; i<N; i++) {
        for(int j=0; j<M; j++) {
            if(vis[i][j] || graph[i][j] == 0) continue;
            cnt++;
            int area = 0;
            vis[i][j] = 1;
            Q.push({i, j});
            while(!Q.empty()) {
                area++;
                auto cur = Q.front(); Q.pop();
                for(int d=0; d<4; d++) {
                    int nx = cur.X + dx[d];
                    int ny = cur.Y + dy[d];

                    if(nx < 0 || nx >= N || ny < 0 || ny >= M) continue;
                    if(vis[nx][ny] || graph[nx][ny] != 1) continue;
                    
                    vis[nx][ny] = 1;
                    Q.push({nx, ny});
                }
            }
            if(area > wide) wide = area;
        }
    }
    cout << cnt << '\n' << wide;
}

dfs로 풀면 `area` 변수 없이 `graph[nx][ny] = graph[cur.X][cur.Y] + 1`로 각 컴포넌트의 너비를 구할 수 있다.

하지만 메모리 측면에서 bfs로 푸는 것이 이득이다.