백준 1926번 그림 (C++)

[Silver I] 그림 - 1926

문제 링크

성능 요약

메모리: 3868 KB, 시간: 56 ms

분류

너비 우선 탐색, 깊이 우선 탐색, 그래프 이론, 그래프 탐색

문제 설명

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

입력

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

출력

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

코드

#include <bits/stdc++.h>
using namespace std;

#define X first
#define Y second

int main(void) {
  int board[502][502];
  int vis[502][502];
  int dx[4] = {1, 0, -1, 0};
  int dy[4] = {0, 1, 0, -1};
  int n, m;
  cin >> n >> m;
  for (int i = 0; i < n; i++) {
    for (int j = 0; j < m; j++) {
      cin >> board[i][j];
    }
  }

  queue<pair<int, int>> Q;

  int size;
  int max_size = 0;
  int count = 0;

  for (int xi = 0; xi < n; xi++) {
    for (int yi = 0; yi < m; yi++) {
      if (board[xi][yi] == 1) {
        size = 0;
        Q.push({xi, yi});
        vis[xi][yi] = 1;
        board[xi][yi] = 0;
        while (!Q.empty()) {
          size++;
          pair<int, int> cur = Q.front();
          Q.pop();
          for (int i = 0; i < 4; i++) {
            int nx = cur.X + dx[i];
            int ny = cur.Y + dy[i];
            if (nx >= n || ny >= m || nx < 0 || ny < 0)
              continue;
            if (board[nx][ny] != 1 || vis[nx][ny] == 1)
              continue;
            Q.push({nx, ny});
            vis[nx][ny] = 1;
            board[nx][ny] = 0;
          }
        }
        if (size > max_size)
          max_size = size;
        count++;
      }
    }
  }
  cout << count << "\n" << max_size;
}

BFS의 시작점을 찾기 위해 2중 for문 안에 BFS 과정을 넣었다. 2중 for문을 도는 중 board에서 시작점 1을 발견하면 BFS의 과정을 수행한 후 다음 시작점을 찾는데 이때 BFS 과정 중 방문했는 곳은 0으로 바꿔서 중복 방문하지 않게 만들었다.

BFS 과정이 몇번 수행되었는지 = size로 생각해 볼 수 있어 size++를 통해 size를 측정하고 시작점 1을 발견한 횟수 = 그림의 개수가 되므로 if 가 수행될 때마다 count++를 진행해 준다