본문 바로가기

[BOJ/C++] 1926번 그림 : BFS

@hyeon.s2023. 1. 12. 19:46

문제 설명

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

입력

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

출력

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

문제 해결

그림의 넓이는 기존의 BFS구현 코드를 이용하면 손쉽게 구할 수 있다. 상하좌우 인접한 부분이 그림이 그려져있고 이미 방문한적이 있는지를 확인한 뒤 그렇지 않으면 상하좌우를 확인하고 front를 pop할때 width값을 1씩 증가해서 최종 그림의 넓이를 구한다. 

그림의 개수는 처음 0,0에서 시작한 뒤 while문을 다 돌면 좌표를 이동해 해당 좌표가 다음 시작점이 될 수 있는지를 확인한다. 시작점이 되기위해서는 방문되지 않고, 그림이 색칠되어 있어야하므로 해당 조건문을 이용해 for문으로 새로운 시작점을 찾을 수 있다. 시작점을 찾으면 앞선 방법처럼 또 while문을 돌면서 넓이를 구하면된다. 각 넓이들은 max값을 이용해 max보다 값이 크면 최댓값을 갱신해주는 방법으로 가장넓은 그림의 넓이를 출력했다. 

코드

#include <iostream>
#include <queue>
using namespace std;
#define X first
#define Y second
int dx[4] = {1,0,-1,0 }; //상하좌우 네 방향
int dy[4] = { 0,1,0,-1 };
int main()
{
	ios::sync_with_stdio(0);
	cin.tie(0);
	int n, m;
	int cnt = 0, max = 0;
	cin >> n >> m;
	bool** arr = new bool* [n];
	bool** vis = new bool* [n];
	for (int i = 0; i < n; i++)
	{
		arr[i] = new bool[m];
		vis[i] = new bool[m];
	}
	for (int i = 0; i < n; i++)
	{
		for (int j = 0; j < m; j++)
		{
			cin >> arr[i][j];
			vis[i][j] = 0;
		}
	}
	for (int i=0; i<n; i++){
		for (int j = 0; j < m; j++)
		{
			int width = 0;
			if (arr[i][j] == 0 || vis[i][j]) continue;
			else
			{
				queue<pair<int, int>> q;
				vis[i][j] = 1;
				q.push({ i,j });
				while (!q.empty()) {
					pair<int, int> cur = q.front(); q.pop(); width++;
					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 (vis[nx][ny] || arr[nx][ny] != 1) continue;
						vis[nx][ny] = 1;
						q.push({ nx,ny });
					}
				}
				cnt++;
				if (max < width) max = width;
			}
		}
	}
	cout << cnt << "\n" << max;
}
hyeon.s
@hyeon.s :: 개발로그
목차