https://www.acmicpc.net/problem/7576
문제
철수의 토마토 농장에서는 토마토를 보관하는 큰 창고를 가지고 있다. 토마토는 아래의 그림과 같이 격자 모양 상자의 칸에 하나씩 넣어서 창고에 보관한다.
창고에 보관되는 토마토들 중에는 잘 익은 것도 있지만, 아직 익지 않은 토마토들도 있을 수 있다. 보관 후 하루가 지나면, 익은 토마토들의 인접한 곳에 있는 익지 않은 토마토들은 익은 토마토의 영향을 받아 익게 된다. 하나의 토마토의 인접한 곳은 왼쪽, 오른쪽, 앞, 뒤 네 방향에 있는 토마토를 의미한다. 대각선 방향에 있는 토마토들에게는 영향을 주지 못하며, 토마토가 혼자 저절로 익는 경우는 없다고 가정한다. 철수는 창고에 보관된 토마토들이 며칠이 지나면 다 익게 되는지, 그 최소 일수를 알고 싶어 한다.
토마토를 창고에 보관하는 격자모양의 상자들의 크기와 익은 토마토들과 익지 않은 토마토들의 정보가 주어졌을 때, 며칠이 지나면 토마토들이 모두 익는지, 그 최소 일수를 구하는 프로그램을 작성하라. 단, 상자의 일부 칸에는 토마토가 들어있지 않을 수도 있다.
입력
상자의 크기를 나타내는 두 정수 M, N이 주어진다. M은 상자의 가로 칸의 수, N은 상자의 세로 칸의 수를 나타낸다.
단, 2 ≤ M,N ≤ 1,000이다.
정수 1은 익은 토마토, 정수 0은 익지 않은 토마토, 정수 -1은 토마토가 들어있지 않은 칸을 나타낸다.
만약, 저장될 때부터 모든 토마토가 익어있는 상태이면 0을 출력해야 하고, 토마토가 모두 익지는 못하는 상황이면 -1을 출력해야 한다.
풀이
- 상자의 토마토가 인접한 네 방향에 영향을 주며, 다 익게 되는 최소 일수를 구하는 문제 -> BFS 문제다.
- 토마토 정보를 입력받을 때 1 (익은 토마토)이면 큐에 넣어준다. (익은 토마토로부터 익기 시작하므로)
- BFS 실행
- 아직 방문하지 않았고, 익지 않은 토마토가 있는 곳을 방문한다.
- day 배열에 해당 칸이 익는데 걸린 일수를 넣는다.
- 토마토가 모두 익은 날을 구해야 하므로 max값에 day배열의 최댓값을 넣어준다.
- 탐색이 모두 끝났는데 아직 익지 않은 토마토가 있고 그 토마토에 방문하지 않았다면 -1을 출력하고 종료.
- 아니면 max값을 출력.
import java.util.*; import java.io.* public class Main { static class Pair { int x; int y; Pair(int x, int y) { this.x = x; this.y = y; } } static int dx[] = {0, 0, 1, -1}; static int dy[] = {1, -1, 0, 0}; public static void main(String[] args) throws IOException { BufferedReader br = new BufferedReader(new InputStreamReader(System.in)); StringTokenizer st = new StringTokenizer(br.readLine()); int m = Integer.parseInt(st.nextToken()); int n = Integer.parseInt(st.nextToken()); int box[][] = new int[n][m]; boolean check[][] = new boolean[n][m]; Queue<Tomato> q = new LinkedList<Tomato>(); for(int i=0; i<n; i++) { st = new StringTokenizer(br.readLine()); for(int j=0; j<m; j++) { box[i][j] = Integer.parseInt(st.nextToken()); if(box[i][j] == 1) { check[i][j] = true; q.add(new Pair(i,j)); } } } int day[][] = new int[n][m]; int max = 0; while(!q.isEmpty()) { Pair p = q.remove(); int x = p.x; int y = p.y; for(int i=0; i<4; i++) { int nx = x + dx[i]; int ny = y + dy[i]; if(nx >=0 && ny >=0 && nx <n && ny < m) if(box[nx][ny] == 0 && check[nx][ny] == false) { check[nx][ny] = true; day[nx][ny] = day[x][y] +1; q.add(new Pair(nx,ny)); if(max < day[nx][ny]) max = day[nx][ny]; } } } for(int i=0; i<n; i++) { for(int j=0; j<m; j++) { if(box[i][j] == 0 && check[i][j] == false){ System.out.println(-1); return; } } } System.out.println(max); } }
'BOJ' 카테고리의 다른 글
[BOJ] 2606. 바이러스 (0) | 2019.05.01 |
---|---|
[BOJ] 4963. 섬의 개수 (0) | 2019.05.01 |
[BOJ] 7569. 토마토 (3차원) (0) | 2019.04.30 |
[BOJ] 11724. 연결 요소의 개수 (0) | 2019.04.29 |
[BOJ] 2309. 일곱 난쟁이 (0) | 2019.04.29 |