본문 바로가기

Computer Science&Engineering/코딩테스트

[ 백준 2573 ] 빙산 (C++) BFS 풀이

# 문제링크

www.acmicpc.net/problem/2573

 

2573번: 빙산

첫 줄에는 이차원 배열의 행의 개수와 열의 개수를 나타내는 두 정수 N과 M이 한 개의 빈칸을 사이에 두고 주어진다. N과 M은 3 이상 300 이하이다. 그 다음 N개의 줄에는 각 줄마다 배열의 각 행을

www.acmicpc.net

BFS 돌리면서 0 만나면 map에 -1 씩 해주고, 0이 아니면 queue 에 넣어서 계속 돌려주면 된다. 그렇게 한 덩이가 2개 이상이면 return 해주면 된다! 

 

# 제출 코드 

 

#include <cstdio>
#include <queue>
#include <cstring>
using namespace std;

int N, M;
int map[300][300];

void input() {
	scanf("%d %d", &N, &M);
	for (int i = 0; i < N; i++) {
		for (int j = 0; j < M; j++) {
			scanf("%d", &map[i][j]);
		}
	}
}

bool visited[300][300];

int xd[] = { 1, -1, 0,0 };
int yd[] = { 0, 0, 1, -1 };

void BFS(int i, int j) {
	queue <pair<int, int >> q;
	q.push({ i,j });
	visited[i][j] = true;

	while (!q.empty()) {
		int y = q.front().first;
		int x = q.front().second;
		q.pop();

		for (int i = 0; i < 4; i++) {
			int ny = y + yd[i];
			int nx = x + xd[i];
			if (ny < 0 || ny >= N)continue;
			if (nx < 0 || nx >= M) continue;
			if (visited[ny][nx])continue;

			if (map[ny][nx] == 0) { // 사방이 바닷물이면 ?
				map[y][x] = (map[y][x] == 0) ? 0 : map[y][x] - 1;
			}
			else { // 빙산이면 
				visited[ny][nx] = true;
				q.push({ ny,nx });
			}
		}
	}
}

int Solve() {
	int ans = 0;
	int ice = 0;
	for ( ; ; ans++) {
		ice = 0;
		for (int i = 0; i < N; i++) {
			for (int j = 0; j < M; j++) {
				if (map[i][j] != 0 && visited[i][j] == false) {
					BFS(i, j);
					ice++; // 빙산 개수 
				}
			}
		}
		if (ice >= 2) return ans;
		if (ice == 0) break;
		memset(visited, 0, sizeof(visited));
	}
	return 0;
}

int main() {
	input();
	printf("%d\n", Solve());
}
반응형