C++ 백준 7576번

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

7576번: 토마토

문제 : 철수의 토마토 농장에서는 토마토를 보관하는 큰 창고를 가지고 있다. 토마토는 아래의 그림과 같이 격자 모양 상자의 칸에 하나씩 넣어서 창고에 보관한다.

창고에 보관되는 토마토들 중에는 잘 익은 것도 있지만, 아직 익지 않은 토마토들도 있을 수 있다. 보관 후 하루가 지나면, 익은 토마토들의 인접한 곳에 있는 익지 않은 토마토들은 익은 토마토의 영향을 받아 익게 된다. 하나의 토마토의 인접한 곳은 왼쪽, 오른쪽, 앞, 뒤 네 방향에 있는 토마토를 의미한다. 대각선 방향에 있는 토마토들에게는 영향을 주지 못하며, 토마토가 혼자 저절로 익는 경우는 없다고 가정한다. 철수는 창고에 보관된 토마토들이 며칠이 지나면 다 익게 되는지, 그 최소 일수를 알고 싶어 한다.

토마토를 창고에 보관하는 격자모양의 상자들의 크기와 익은 토마토들과 익지 않은 토마토들의 정보가 주어졌을 때, 며칠이 지나면 토마토들이 모두 익는지, 그 최소 일수를 구하는 프로그램을 작성하라. 단, 상자의 일부 칸에는 토마토가 들어있지 않을 수도 있다.

 

익은 토마토가 있으면 그 주위의 상하좌우 토마토들이 하루 뒤에 익게 된다. 창고 안의 모든 토마토들이 익는데 걸리는 날짜를 구하는데 만약 토마토가 하나라도 안 익으면 -1을 출력한다.

익은 토마토들을 큐에 넣고 BFS를 사용하는데 이 큐에있는(익은) 토마토 상하좌우에 안 익고 아직 방문되지 않은 토마토가 있으면 걔를 익히고 날짜를 증가시키고 걔도 큐에 넣는 걸 반복하면 된다.

모든 과정이 끝났는데 아직 안익었고 방문되지 않은 토마토가 있으면 -1을 출력하면 된다!!

#include<iostream>
#include<queue>
#include<algorithm>
using namespace std;
int a[1001][1001];
int dx[] = { 0, -1, 0, 1 };
int dy[] = { -1, 0, 1, 0 };
int d[1001][1001]; // 토마토 익은 날짜
int main() {
	int m, n; // m 가로(열), n 세로(행)
	cin >> m >> n;
	queue<pair<int, int>> q;
	for (int i = 0; i < n; i++) {
		for (int j = 0; j < m; j++) {
			cin >> a[i][j];
			d[i][j] = -1; // 아직 방문하지 않은 토마토
			if (a[i][j] == 1) {
				q.push(make_pair(i, j)); // 익은 토마토가 있으면 큐에 넣기
				d[i][j] = 0; //원래 익은 친구들이니까 0
			}
		}
	}
	while (!q.empty()) {
	int x = q.front().first; // 첫번째로 들어간 익은 토마토 x 좌표
	int y = q.front().second; // 첫번째로 들어간 익은 토마토의 y 좌표
	q.pop();
		for (int i = 0; i < 4; i++) {
			int nx = x + dx[i];
			int ny = y + dy[i];
			if (nx >= 0 && nx < n && ny >= 0 && ny < m) {
				if (a[nx][ny] == 0 && d[nx][ny] == -1) {
					d[nx][ny] = d[x][y] + 1; // 날짜 증가
					q.push(make_pair(nx, ny));
				}
			}
		}
	}
	int result = 0;
	for (int i = 0; i < n; i++) {
		for (int j = 0; j < m; j++) {
			result = max(result, d[i][j]);
		}
	}
	for (int i = 0; i < n; i++) {
		for (int j = 0; j < m; j++) {
			if (a[i][j] == 0 && d[i][j] == -1) { // 안익고 방문된적 없으면 -1
				result = -1;
			}
		}
	}
	cout << result;
}