BOJ/BFS\DFS

[C언어 / 백준 - 4963] 섬의 개수 (BFS)

JWonK 2021. 5. 14. 23:14
728x90
반응형

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

 

4963번: 섬의 개수

입력은 여러 개의 테스트 케이스로 이루어져 있다. 각 테스트 케이스의 첫째 줄에는 지도의 너비 w와 높이 h가 주어진다. w와 h는 50보다 작거나 같은 양의 정수이다. 둘째 줄부터 h개 줄에는 지도

www.acmicpc.net

문제

정사각형으로 이루어져 있는 섬과 바다 지도가 주어진다. 섬의 개수를 세는 프로그램을 작성하시오.

한 정사각형과 가로, 세로 또는 대각선으로 연결되어 있는 사각형은 걸어갈 수 있는 사각형이다. 

두 정사각형이 같은 섬에 있으려면, 한 정사각형에서 다른 정사각형으로 걸어서 갈 수 있는 경로가 있어야 한다. 지도는 바다로 둘러싸여 있으며, 지도 밖으로 나갈 수 없다.

입력

입력은 여러 개의 테스트 케이스로 이루어져 있다. 각 테스트 케이스의 첫째 줄에는 지도의 너비 w와 높이 h가 주어진다. w와 h는 50보다 작거나 같은 양의 정수이다.

둘째 줄부터 h개 줄에는 지도가 주어진다. 1은 땅, 0은 바다이다.

입력의 마지막 줄에는 0이 두 개 주어진다.

출력

각 테스트 케이스에 대해서, 섬의 개수를 출력한다.

 

BFS를 이용하여 풀 수 있는 문제이지만 아주 작은 약간의(?) 변형이 있는데 대각선까지 이동할 수 있다는 점이다.

이 부분만 고려하여 짜면 기존 BFS와 똑같은 문제이다.

 

1이 육지, 0이 바다이므로 지도를 훑어가며 육지일 경우에는 큐에 넣어주어 그 큐에서 갈 수 있는 모든 육지는 또 다시 넣어준다. 

그리고 이제 큐가 비어있다면 연결되어있는 육지는 모두 다 갔던 곳이므로 방문했다는 표시를 남겨준 후 다시 지도를 훑어주며 방문하지 않는 육지는 똑같이 큐에 넣어준다. 이 과정을 통해 육지의 개수를 파악할 수 있다.

#define _CRT_SECURE_NO_WARNINGS
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <math.h>
#include <stdbool.h>
#define MAX_SIZE 51
struct node {
	int x;
	int y;
};

struct node queue[2500];
int head = 0, tail = 0;

void Enque(int x, int y) {
	struct node temp;
	temp.x = x;
	temp.y = y;
	queue[tail] = temp;
	tail = (tail + 1) % 2500;
}

struct node Deque() {
	struct node temp = queue[head];
	head = (head + 1) % 2500;
	return temp;
}

int Is_Empty() {
	if (head == tail) return 0;
	else return 1;
}

int dx[8] = {1,-1,0,0,-1,-1,1,1};
int dy[8] = {0,0,1,-1,-1,1,-1,1};

int visited[MAX_SIZE][MAX_SIZE];
int map[MAX_SIZE][MAX_SIZE];
int N, M,cnt;

void BFS() {
	int nextX, nextY;

	while (Is_Empty()) {
		struct node ps = Deque();

		for (int i = 0; i < 8; i++) {
			nextX = ps.x + dx[i];
			nextY = ps.y + dy[i];

			if (0 <= nextX && nextX < M && 0 <= nextY && nextY < N) {
				if (!visited[nextX][nextY] && map[nextX][nextY]) {
					Enque(nextX, nextY);
					visited[nextX][nextY] = 1;
				}
			}
		}
	}
}

int main() {
	while (1) {
		scanf("%d %d", &N, &M);
		if (N == 0 && M == 0) break;

		cnt = 0;
		memset(map, 0, sizeof(map));
		memset(visited, 0, sizeof(visited));
		for (int i = 0; i < M; i++) 
			for (int j = 0; j < N; j++) 
				scanf("%d", &map[i][j]);
			
		for (int i = 0; i <M; i++) {
			for (int j = 0; j < N; j++) {
				if (!visited[i][j] && map[i][j]) {
					Enque(i, j);
					visited[i][j] = 1;
					BFS();
					cnt++;
				}
			}
		}
		printf("%d\n", cnt);
	}

	return 0;
}
728x90
반응형