[백준(BOJ) 7576번] 토마토 (C++)

문제 링크

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

 

문제 정보

입력

첫 줄에는 상자의 크기를 나타내는 두 정수 M,N이 주어진다. M은 상자의 가로 칸의 수, N은 상자의 세로 칸의 수를 나타낸다. 단, 2 ≤ M,N ≤ 1,000 이다. 둘째 줄부터는 하나의 상자에 저장된 토마토들의 정보가 주어진다. 즉, 둘째 줄부터 N개의 줄에는 상자에 담긴 토마토의 정보가 주어진다. 하나의 줄에는 상자 가로줄에 들어있는 토마토의 상태가 M개의 정수로 주어진다. 정수 1은 익은 토마토, 정수 0은 익지 않은 토마토, 정수 -1은 토마토가 들어있지 않은 칸을 나타낸다.

토마토가 하나 이상 있는 경우만 입력으로 주어진다.

출력

여러분은 토마토가 모두 익을 때까지의 최소 날짜를 출력해야 한다. 만약, 저장될 때부터 모든 토마토가 익어있는 상태이면 0을 출력해야 하고, 토마토가 모두 익지는 못하는 상황이면 -1을 출력해야 한다.

 

풀이

 이 문제는 BFS의 응용 문제로 시작점이 여러 개일 때 문제를 해결하는 방법이다. 시작점을 모두 큐에 넣어버리고 BFS를 돌려서 해결한다!

 

소스 코드

#include <bits/stdc++.h>
using namespace std;

#define X first
#define Y second

int board[1002][1002];
int dist[1002][1002]; // 익은 날짜 저장할 배열
int n, m;
int dx[4] = { 1, 0, -1, 0 };
int dy[4] = { 0, 1, 0, -1 }; 

int main(void) {
	ios::sync_with_stdio(0);
	cin.tie(0);
	
	cin >> m >> n;
	
	queue<pair<int, int>> Q;
	
	for (int i = 0; i < n; i++) {
		for (int j = 0; j < m; j++) {
			cin >> board[i][j];
			
			if (board[i][j] == 1) {
				Q.push({i, j}); // 익은 토마토(시작 좌표)를 큐에 넣어버린다.
			}
			
			if (board[i][j] == 0) {
				dist[i][j] = -1; // 익지 않은 토마토의 익은 날짜를 -1로 저장한다. (익은 토마토와 토마토가 없는 칸은 0으로 초기화 되어 있음!)
			}
		}
	}
	
	while (!Q.empty()) {
		auto cur = Q.front();
		Q.pop();
		
		for (int dir = 0; dir < 4; dir++) {
			int nx = cur.X + dx[dir];
			int ny = cur.Y + dy[dir];
			
			if (nx < 0 || nx >= n || ny < 0 || ny >= m) {
				continue;
			}
			if (dist[nx][ny] >= 0) {
				continue; // 익은 토마토, 토마토가 없는 칸
			}
			
			dist[nx][ny] = dist[cur.X][cur.Y] + 1;
			Q.push({nx, ny});
		}
	}
	
	int ans = 0;
	for (int i = 0; i < n; i++) {
		for (int j = 0; j < m; j++) {
			if (dist[i][j] == -1) {
				cout << "-1";
				return 0;
			}
			ans = max(ans, dist[i][j]);
		}
	}
	
	cout << ans;
	
	return 0;
}