C++ 백준 2667번

백준 2667번 www.acmicpc.net/problem/2667

2667번: 단지번호붙이기

문제 : <그림 1>과 같이 정사각형 모양의 지도가 있다. 1은 집이 있는 곳을, 0은 집이 없는 곳을 나타낸다. 철수는 이 지도를 가지고 연결된 집의 모임인 단지를 정의하고, 단지에 번호를 붙이려 한다. 여기서 연결되었다는 것은 어떤 집이 좌우, 혹은 아래위로 다른 집이 있는 경우를 말한다. 대각선 상에 집이 있는 경우는 연결된 것이 아니다. <그림 2>는 <그림 1>을 단지별로 번호를 붙인 것이다. 지도를 입력하여 단지수를 출력하고, 각 단지에 속하는 집의 수를 오름차순으로 정렬하여 출력하는 프로그램을 작성하시오.

상하좌우 옆으로 붙어있는 집들끼리 연결이 된 거를 단지라고 부르고 이 단지수와 그 단지의 집 수를 구하는 문제이다.

bfs나 dfs를 이용하여 연결돼있는 집들을 탐색하고 단지 하나가 연결되면 새로운 단지를 연결하는데 이때 그 전의 단지 개수를 vector에 push 해놓고 새로운 단지 개수를 받으면 맨 마지막의 벡터 원소 개수가 단지 개수일 것이고 원소마다 저장된 값이 각 단지의 집 개수일 것이다!!

나는 큐 쓰기 싫어서 dfs를 사용했는데 bfs를 사용해도 된다.

#define _CRT_SECURE_NO_WARNINGS
#include<cstdio>
#include<vector>
#include<algorithm>
#include<iostream>
using namespace std;
int map[26][26];
bool check[26][26];
vector<int> v;
int n;
int cnt = 0;
int dir[4][2] = { {-1,0}, {0,1},{1,0},{0,-1} }; // 상하좌우 이동
void dfs(int x, int y) {
	for (int i = 0; i < 4; i++) {
		int nx = x + dir[i][0];
		int ny = y + dir[i][1];

		if (nx >= 0 && ny >= 0 && nx <= n && ny <= n) {
			if (check[nx][ny] == false && map[nx][ny] == 1) {
				check[nx][ny] = true;
				cnt++;
				dfs(nx, ny);
			}
		}
	}
}
int main() {
	scanf("%d",&n);
	for (int i = 0; i < n; i++) {
		for (int j = 0; j < n; j++) {
			scanf("%1d", &map[i][j]); // 정수 한개씩 입력받음
		}
	}
	for (int i = 0; i < n; i++) {
		for (int j = 0; j < n; j++) {
			if (check[i][j] == false && map[i][j] == 1) {
				check[i][j] = true;
				cnt++;
				dfs(i, j);
				v.push_back(cnt); // 단지개수 push
				cnt = 0;
			}
		}
	}
	cout << v.size() << '\n'; // v의 size가 단지 개수
	sort(v.begin(), v.end());
	for (int i = 0; i < v.size(); i++) {
		cout << v[i] << '\n';
	}
}