알고리즘

백준 27737번 버섯 농장 (JAVA)

박카스마시며코딩 2023. 6. 13. 19:48

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

 

27737번: 버섯 농장

첫 번째 줄에 $N$, $M$, $K$가 공백으로 구분되어 주어진다. 두 번째 줄부터 $N$개의 줄에 나무판의 각 칸의 상태가 공백으로 구분되어 주어진다. 버섯이 자랄 수 있는 칸은 0, 버섯이 자랄 수 없는 칸

www.acmicpc.net

 

저는 BFS를 통해 문제를 해결했습니다.

BFS를 통해 인접한 빈칸들을 확인하고 버섯이 몇개 필요하는지를 체크하고 이를 전부 더해주었습니다.

혹시 75% 에서 틀렸습니다가 나온다면 문제의 마지막 문구를 잘 보시기 바랍니다. 

즉, 전체가 버섯이 자랄 수 없는 칸이라면 IMPOSSIBLE을 출력해야합니다.

 

package BOJ.bfs;

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.LinkedList;
import java.util.Queue;
import java.util.StringTokenizer;
import java.util.function.Function;

public class BOJ_27737 {
    private static final int BLOCK = 1;
    private static final int EMPTY = 0;
    private static final String IMPOSSIBLE = "IMPOSSIBLE";
    private static final String POSSIBLE = "POSSIBLE";
    public static void main(String[] args) throws Exception {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());
        Function<String,Integer> stoi = Integer::parseInt;
        int n = stoi.apply(st.nextToken());
        int m = stoi.apply(st.nextToken());
        int k = stoi.apply(st.nextToken());
        int[][] map = new int[n][n];
        for(int i = 0 ; i < n ; i++){
            st = new StringTokenizer(br.readLine());
            for(int j = 0 ; j < n ; j++){
                map[i][j] = stoi.apply(st.nextToken());
            }
        }
        int result = cal(map,n,k);

        if(result > m || m == 0 || result == 0){
            System.out.println(IMPOSSIBLE);
        }else{
            System.out.println(POSSIBLE);
            System.out.println(m - result);
        }
    }

    private static int cal(int[][] map, int n, int k) {
        boolean[][] visited = new boolean[n][n];
        int cnt = 0;
        for(int i = 0 ; i < n ; i++){
            for(int j = 0 ; j < n ; j++){
                if(!visited[i][j] && map[i][j] == EMPTY){
                    cnt += bfs(i,j,map,visited,n,k);
                }
            }
        }
        return cnt;
    }
    private static final int[] DY = {-1,0,1,0};
    private static final int[] DX = {0,1,0,-1};
    private static int bfs(int y, int x, int[][] map, boolean[][]visited, int n, int k) {
        Queue<int[]> q = new LinkedList<>();
        q.offer(new int[]{y,x});
        visited[y][x] = true;
        int cnt = 0;
        while(!q.isEmpty()){
            int[] now = q.poll();
            cnt++;
            for(int i = 0 ; i < 4 ; i++){
                int ny = now[0] + DY[i];
                int nx = now[1] + DX[i];
                if(ny >= 0 && ny < n && nx >= 0 && nx < n && !visited[ny][nx] && map[ny][nx] == EMPTY){
                    visited[ny][nx] = true;
                    q.offer(new int[]{ny,nx});
                }
            }
        }
        int result = cnt / k;
        if(cnt % k != 0){
            result++;
        }
        return result;
    }
}