백준- 7576번

문딤·2022년 8월 3일

토마토

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

풀이 생각

상하좌우? bfs
벗어나지 않게 범위지정 할 것.

소스코드


import java.io.*;
import java.util.*;

class ArMain {
    static int N;
    static int M;
    static int[][] box;
    static int[] dx = {1, -1, 0, 0};
    static int[] dy = {0, 0, 1, -1};

    public static void main(String[] args) throws Exception {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());
        N = Integer.parseInt(st.nextToken());
        M = Integer.parseInt(st.nextToken());

        box = new int[M][N];

        for(int i=0; i<M; i++) {
            st = new StringTokenizer(br.readLine());

            for(int j=0; j<N; j++) {
                box[i][j] = Integer.parseInt(st.nextToken());
            }
        }

        bfs();
    }

Dot 클래스

 static class Dot {
        int x;
        int y;
        int day;

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

bfs

    static void bfs() {
        Queue<Dot> q = new LinkedList<Dot>();
        //제너릭에 Dot 클래스가 들어감 .
        //q= >(x,y,day)가 들어감.
        int day = 0;

        // 토마토가 있는 좌표 찾아서 Queue에 넣기
        for(int i=0; i<M; i++) {
            for(int j=0; j<N; j++) {
                if(box[i][j] == 1)
                    
                    q.offer(new Dot(i, j, 0));
            
            }
        }
        // bfs 시작
        while(!q.isEmpty()) {
            Dot dot = q.poll();
            day = dot.day;
            //처음 들어간애 poll
            //익은 애들은 중복방지용으로 큐에 넣고 주변 애들을 검색

            for(int i=0; i<4; i++) {
                int nx = dot.x + dx[i]; //3 +(1,-1,0,0)
                int ny = dot.y + dy[i]; //5(0,0 ,-1 ,1)

                //다음날 이면 익는 토마토 위치

                if(0 <= nx && nx < M && 0 <= ny && ny < N) {
                    //벗어나지 않는다면
                    if(box[nx][ny] == 0) {
                        //0인 애들 익혀주기
                        box[nx][ny] = 1;
                        q.add(new Dot(nx, ny, day+1));
                        //익었으니깐 q에 넣구 카운트
                        //day 카운트
                    }
                }
            }
        }

        // 토마토가 다 익었는지 확인
        if(checkTomato())
            System.out.println(day);
        else
            System.out.println(-1);
    }

    // box 배열에 0이 남아있다면 false, 아니면 true
    static boolean checkTomato() {
        for(int i=0; i<M; i++) {
            for(int j=0; j<N; j++) {
                if(box[i][j] == 0)
                    return false;
            }
        }

        return true;
    }
}