문제
<그림 1>과 같이 정사각형 모양의 지도가 있다. 1은 집이 있는 곳을, 0은 집이 없는 곳을 나타낸다.
철수는 이 지도를 가지고 연결된 집의 모임인 단지를 정의하고, 단지에 번호를 붙이려 한다.
여기서 연결되었다는 것은 어떤 집이 좌우, 혹은 아래위로 다른 집이 있는 경우를 말한다. 대각선상에 집이 있는 경우는 연결된 것이 아니다.
<그림 2>는 <그림 1>을 단지별로 번호를 붙인 것이다.
지도를 입력하여 단지수를 출력하고, 각 단지에 속하는 집의 수를 오름차순으로 정렬하여 출력하는 프로그램을 작성하시오.
입력
첫 번째 줄에는 지도의 크기 N(정사각형이므로 가로와 세로의 크기는 같으며 5≤N≤25)이 입력되고, 그 다음 N줄에는 각각 N개의 자료(0혹은 1)가 입력된다.
출력
첫 번째 줄에는 총 단지수를 출력하시오. 그리고 각 단지내 집의 수를 오름차순으로 정렬하여 한 줄에 하나씩 출력하시오.
해결 방법
정말 정말 힘들었던 문제다 ㅎㅎ
dfs문제를 2개나 풀고 그 다음으로 선택한 dfs문제라 바로 풀 수 있을 줄 알았지^^
근데 놀라운 사실은 이게 초등학교 1학년들이 출전하는 대회 문제라는 사실~ (눈물...)
matrix를 입력받을 때 %1d로 matrix[i][j]를 입력받았는데,
지도의 크기가 최대 25이므로 int나 lon long의 타입으로 입력받으면 안된다고 한다.
가능하면 문자열로 입력받는 것이 좋다고 하여
%s로 matrix[i]를 받게끔 변경.
또 한가지 주의할 점은 지도의 크기가 25라고 해서 단지의 수도 25개인 것은 아니다.
한 단지에 수백개의 집이 있을 수도 있고, 지도에 수백개의 단지가 있을 수도 있기 때문이라고 한다.
그럼 지도의 크기가 25라는 것은 왜 준거야..?
그리고..... 내가 질문 게시판에 결국 글을 쓸 정도로 모든 반례와 예시가 비쥬얼 스튜디오에서는 통과되었지만 백준에서만 틀렸다고 뜬 이유...........
나의 생각 과정
- arr[]의 크기를 500으로 넉넉하게 설정한 뒤에 dfs를 돌림
- 예를 들어 arr[0], arr[1], arr[2]의 값이 채워지면 인덱스 0부터 500까지 arr[]를 버블소트를 사용하여 오름차순 정렬
- 마지막에 printf문에서 arr[]의 값이 0이 아닌 것만 출력하게 함,
즉 나는 3개 밖에 안되는 값을 출력하기 위해 arr[497], arr[498], arr[499]의 값을 출력하고 있었던 것^^
결론!
버블소트에서 애초에 i의 초기값을 n - 1이 아닌, count - 1로 설정한다면
arr[0], arr[1], arr[2] 끼리만 오름차순 정렬이 되어 시간과 메모리를 잡아먹지 않음(내 질문에 댓글 달아주신 분 정말 감사합니다,,)
아래는 내 코드(구글링 많이 했지만 그래도 큰 거 하나 배워가는 의미있는 시간이었다)
💡 내 코드(C)
// [2667] 단지번호붙이기
// https://www.acmicpc.net/problem/2667
#include <stdio.h>
char matrix[30][30];
int visit[30][30] = { 0, };
int arr[500];
int n;
int count = 0;
void dfs(int x, int y);
int main()
{
int temp;
scanf("%d", &n);
for (int i = 0; i < n; i++) {
scanf("%s", matrix[i]);
}
for (int i = 0; i < n; i++) {
for (int j = 0; j < n; j++) {
if (matrix[i][j] == '1' && !visit[i][j]) {
dfs(i, j);
count++;
}
}
}
// bubble sort, i = count - 1
for (int i = count - 1; i > 0; i--) {
for (int j = 0; j < i; j++) {
if (arr[j] > arr[j + 1]) {
temp = arr[j];
arr[j] = arr[j + 1];
arr[j + 1] = temp;
}
}
}
printf("%d\n", count);
for (int i = 0; i < count; i++) {
if (arr[i])
printf("%d\n", arr[i]);
}
return 0;
}
void dfs(int x, int y)
{
visit[x][y] = 1;
arr[count] += 1;
if (x - 1 >= 0 && !visit[x - 1][y] && matrix[x - 1][y] =='1') // down
dfs(x - 1, y);
if (x + 1 < n && !visit[x + 1][y] && matrix[x + 1][y] =='1') // up
dfs(x + 1, y);
if (y - 1 >= 0 && !visit[x][y - 1] && matrix[x][y - 1] =='1') // left
dfs(x, y - 1);
if (y + 1 < n && !visit[x][y + 1] && matrix[x][y + 1] =='1') // right
dfs(x, y + 1);
}
'Baekjoon > DFS' 카테고리의 다른 글
[11724] 연결 요소의 개수 (0) | 2020.06.23 |
---|---|
[4963] 섬의 개수(C++) (0) | 2020.05.19 |
[2668] 숫자 고르기(C++) (0) | 2020.01.12 |
[1012] 유기농 배추 (0) | 2019.07.22 |
[2606] 바이러스 (0) | 2019.07.14 |