[백준 1926] 그림(C++)

https://www.acmicpc.net/problem/1926

1926번: 그림

문제

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

입력

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

출력

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

내 풀이

도화지에서 새로운 그림(?)을 만날때마다 BFS를 통해 그림의 넓이를 구해준다.

#include <iostream>
#include <vector>
#include <queue>
using namespace std;
 
int main()
{
	int width, height;
	cin >> height >> width;
 
	vector<vector<int>> map = vector<vector<int>>(height, vector<int>(width, 0));
	vector<vector<bool>> visited = vector<vector<bool>>(height, vector<bool>(width, false));
	queue<pair<int, int>> q; // <height, width>
 
	int dw[4] = { 1, 0, -1, 0 };
	int dh[4] = { 0, 1, 0, -1 };
 
	for (int h = 0; h < height; ++h)
	{
		for (int w = 0; w < width; ++w)
		{
			cin >> map[h][w];
		}
	}
 
	int count = 0;
	int size = 0;
	int max = 0;
 
	for (int h = 0; h < height; ++h)
	{
		for (int w = 0; w < width; ++w)
		{
			if (map[h][w] == 0 || visited[h][w])
				continue;
 
			q.push({h, w});
 
			++count;
			size = 0;
 
			while (!q.empty())
			{
				int h = q.front().first;
				int w = q.front().second;
 
				q.pop();
 
				if (visited[h][w])
					continue;
 
				visited[h][w] = true;
				++size;
 
				for (int i = 0; i < 4; ++i)
				{
					int nh = h + dh[i];
					int nw = w + dw[i];
 
					if (nh < 0 || nh >= height || nw < 0 || nw >= width 
						|| map[nh][nw] == 0 || visited[nh][nw])
					{
						continue;
					}
 
					q.push({nh, nw});
				}
			}
 
			max = max < size ? size : max;
		}
	}
 
	cout << count << '\\n' << max;
 
	return 0;
 
}