Menu

Subscribe

[JAVA/백준] DFS와 BFS: 토마토

HEESOO on Boj

👀 문제

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

👊 도전

1. 설계

  1. 토마토가 익는 최소 일수를 구해야하므로 BFS를 사용한다.
  2. 익은 토마토의 위치를 큐에 넣은 후, 하나씩 빼서 덜 익은 토마토로 상하좌우 이동(큐에 삽입)한다.

2. 구현 (성공 코드)

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77

import java.util.LinkedList;
import java.util.Queue;
import java.util.Scanner;

/**
 * @author HEESOO
 *
 */
class Node{
	int x;
	int y;
	public Node(int x, int y) {
		this.x=x;
		this.y=y;
	}
}
public class Main {
	static int m, n, day, zero;
	static int[][] map;
	static boolean[][] visit;
	public static void bfs() {
		Queue<Node> q=new LinkedList<>();
		for(int i=0;i<n;i++) {
			for(int j=0;j<m;j++) {
				if(map[i][j]==1) {
					visit[i][j]=true;
					q.offer(new Node(i,j));
				}
			}
		}
		
		int cnt=0;;
		while(!q.isEmpty()) {
			Node node=q.poll();
			cnt=map[node.x][node.y]+1;
			int[] dotX= {0,0,-1,1};
			int[] dotY= {-1,1,0,0};
			for(int i=0;i<4;i++) {
				int xx=node.x+dotX[i];
				int yy=node.y+dotY[i];
				if(0<=xx&&xx<n&&0<=yy&&yy<m) 
					if(map[xx][yy]!=-1&&!visit[xx][yy]) {
						zero--;
						visit[xx][yy]=true;
						map[xx][yy]=cnt;
						q.offer(new Node(xx,yy));
					}
			}
		}
		day=cnt-2;
	}
	public static void main(String[] args) {
		// TODO Auto-generated method stub
		Scanner scan = new Scanner(System.in);
		m=scan.nextInt();
		n=scan.nextInt();
		map=new int[n][m];
		visit=new boolean[n][m];
		zero=0;
		for(int i=0;i<n;i++) {
			for(int j=0;j<m;j++) {
				map[i][j]=scan.nextInt();
				if(map[i][j]==0) zero++;
			}
		}
		
		bfs();
		
		if(zero!=0)
			System.out.println("-1");
		else
			System.out.println(day);
		
	}
}

3. 결과

실행결과 🤟 성공 🤟

4. 설명

  1. BFS를 구현한다
    • 최소 일수를 구해야하므로 BFS를 사용한다.
    • 큐에 익은 토마토의 위치를 클래스 Node를 통해 모두 저장한다. 하나만 저장하고 while문을 시작하지 않도록 한다. 익은 토마토가 여러 개일 경우 중간 지점에서 만날 수 있으므로 계속 확인해야하기 때문이다.
    • zero 변수를 두어 덜 익은 토마토가 몇 개인지 확인한다. 토마토가 모두 익지 못할 경우를 확인하기 위함이다.
    • bfs()의 while문을 통해 현재 위치에서 상하좌우로 움직일 수 있는 곳을 큐에 다시 넣는다. 이때 방문할 수 있는 곳에 현재 노드 값(map[i][j])+1을 하여 일수를 저장한다.
    • 마지막 while문을 빠져나오면 cnt에는 최종 일수가 저장되어있다. 이때 주의해야할 점은, map에서 1부터 시작하므로 -1을 해야하고(0부터 시작해야함), while문에 들어갔을때 cnt 사용을 위해 cnt+1을 하므로 다시 -1해줘야한다(마지막 노드는 다음 노드에 cnt+1을 저장할 필요가 없음).
    • 다시 main()으로 돌아와 zero변수를 확인한 후 zero==0이면 모두 익은 것이므로 day를 리턴하고, zero!=0이라면 토마토가 모두 익지 못하는 상황이므로 -1을 출력한다.

👏 해결 완료!