[Java] 백준 1926번 그림

이 글은 백준 1926번 그림을 풀이한다. 코드는 Java로 구현하였다.

문제

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

입력

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

예제 입력 1

6 5
1 1 0 1 1
0 1 1 0 0
0 0 0 0 0
1 0 1 1 1
0 0 1 1 1
0 0 1 1 1

출력

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

예제 출력 1

4
9

제한

• 시간: 2초
• 메모리: 128MB

풀이

접근

어려워보이지만, BFS의 기본 예제 문제이다.

 

이 문제를 해결하기 위해 우리가 해야할 일은 2개다.

1. 상하좌우로 연결된 그림의 크기를 알아내기
2. 도화지에 있는 모든 그림을 찾아내기

 

1번의 경우는 큐에서 pop을 몇 번 하는지만 세면 끝이다.

2번의 경우는 이중 for문을 돌면서 BFS의 시작점이 될 수 있는 곳을 찾으면 된다.

먼저, (0,0)은 그림 칸이고 아직 방문을 안했으니 여기서 BFS를 시작하면 크기가 4인 그림이 찾아진다.

BFS를 돌고 나면 방문 표시도 적절하게 남게 된다.

그 후로 (0,1)을 체크하는데 (0,1)은 그림 칸이지만 이미 앞에서 방문을 했으니 패스한다.

다음은 (0,2)인데 여기는 그림 칸이 아니기 때문에 패스한다.

그 다음 (0,3)은 그림 칸이고 아직 방문하지 않았으니 새롭게 BFS를 시작한다.

 

이와 같이 이중 for문으로 각 칸이 BFS의 시작점이 될 수 있는지를 체크해주면 도화지에 있는 모든 그림을 찾아낼 수 있다.

구현

코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.LinkedList;
import java.util.Queue;
import java.util.StringTokenizer;

class Pair {

  int X, Y;

  Pair(int X, int Y) {
    this.X = X;
    this.Y = Y;
  }
}

public class baekjoon1926 {

  static int n, m;
  static int[][] board;
  static boolean[][] vis;
  static int[] dx, dy;
  static Queue<Pair> Q;

  public static void main(String[] args) throws IOException {
    BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
    StringTokenizer st = new StringTokenizer(br.readLine());
    n = Integer.parseInt(st.nextToken());
    m = Integer.parseInt(st.nextToken());

    board = new int[n][m];
    vis = new boolean[n][m];
    Q = new LinkedList<>();

    dx = new int[] { 1, 0, -1, 0 };
    dy = new int[] { 0, 1, 0, -1 };
    // 배열 값 세팅
    for (int i = 0; i < n; i++) {
      st = new StringTokenizer(br.readLine());
      for (int j = 0; j < m; j++) {
        board[i][j] = Integer.parseInt(st.nextToken());
      }
    }

    // 시작
    int count = 0; // 그림의 개수
    int area = 0; // 넓이
    int maxArea = 0; // 최대 넓이
    for (int i = 0; i < n; i++) {
      for (int j = 0; j < m; j++) {
        // 0이거나 방문한 적이 없으면 생략
        if (board[i][j] == 0 || vis[i][j]) {
          continue;
        }
        count++;
        Q.offer(new Pair(i, j));
        vis[i][j] = true;
        area = 0;
        while (!Q.isEmpty()) {
          Pair cur = Q.poll();
          area++; // 넓이 1 증가
          for (int dir = 0; dir < 4; dir++) {
            int nx = cur.X + dx[dir];
            int ny = cur.Y + dy[dir];
            if (nx < 0 || nx >= n || ny < 0 || ny >= m) continue;
            if (board[nx][ny] != 1 || vis[nx][ny]) continue;
            vis[nx][ny] = true;
            Q.offer(new Pair(nx, ny));
          }
        }
        if (area > maxArea) {
          maxArea = area;
        }
      }
    }

    System.out.println(count);
    System.out.println(maxArea);
  }
}