본문 바로가기
Problem Solving/Baekjoon

[백준] 7576 토마토 - Graph Theory / Java

by graycode 2022. 5. 16.

 문제 링크

 

7576번: 토마토

첫 줄에는 상자의 크기를 나타내는 두 정수 M,N이 주어진다. M은 상자의 가로 칸의 수, N은 상자의 세로 칸의 수를 나타낸다. 단, 2 ≤ M,N ≤ 1,000 이다. 둘째 줄부터는 하나의 상자에 저장된 토마토

www.acmicpc.net

 

 풀이 과정

기본적인 접근 방식은 BFS를 활용한 미로 탐색과 유사하다.

Queue 와 4 방향 좌표 이동에 필요한 dx, dy 배열을 활용하며 이 문제의 경우 방문 여부를 체크할 배열은 필요가 없다.

 

탐색할 위치를 동시 다발적으로 Queue 에 삽입해 탐색하고 4 방향 이동 가능 여부를 체크하는 과정에서

0 이 아닌, 즉 아직 익지 않은 토마토의 좌표는 건너뛰며 탐색을 진행한다.

 

이때 이동할 때 마다 이전 인덱스의 값보다 1 증가시킨 값을 이동할 위치에 저장해 날짜의 증가 값을 구현한다.

 

끝으로 탐색이 끝난 상자를 다시 한번 for 문을 통해 확인한다.

각 인덱스 별 저장된 값 중 가장 큰 값이 모든 토마토가 익는데 필요한 최소 일수가 되며,

이 중 하나라도 익지 않은 토마토가 존재할 시 -1 을 반환한다.

 

 풀이 코드

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.LinkedList;
import java.util.Queue;

public class Main {

	static int m;
	static int n;
	static int[][] box;
	static Queue<Loc> q;
	static int[] dy = { -1, 1, 0, 0 };
	static int[] dx = { 0, 0, -1, 1 };

	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));

		String[] s = br.readLine().split(" ");
		m = Integer.parseInt(s[0]);
		n = Integer.parseInt(s[1]);

		box = new int[n][m];
		q = new LinkedList<>();
		for (int i = 0; i < n; i++) {
			s = br.readLine().split(" ");
			for (int j = 0; j < m; j++) {
				box[i][j] = Integer.parseInt(s[j]);

				if (box[i][j] == 1)
					q.offer(new Loc(i, j));
			}
		}

		bw.write(bfs() + "\n");
		bw.flush();
	}

	private static int bfs() {
		while (!q.isEmpty()) {
			Loc now = q.poll();
			for (int i = 0; i < 4; i++) {
				int nextY = now.y + dy[i];
				int nextX = now.x + dx[i];

				if (nextY < 0 || nextX < 0 || nextY >= n || nextX >= m)
					continue;
				if (box[nextY][nextX] != 0)
					continue;

				box[nextY][nextX] = box[now.y][now.x] + 1;
				q.offer(new Loc(nextY, nextX));
			}
		}

		int day = 0;
		for (int i = 0; i < n; i++) {
			for (int j = 0; j < m; j++) {
				if (box[i][j] == 0) {
					day = -1;
					return day;
				}

				day = Math.max(day, box[i][j]);
			}
		}

		return day - 1;
	}

	private static class Loc {
		int y;
		int x;

		public Loc(int y, int x) {
			this.y = y;
			this.x = x;
		}
	}

}

댓글