백준 14502 - 연구소 (Java)

장준혁·2024년 3월 31일

coding-test

목록 보기
1/21

🔍 문제 설명



📌 입력

첫째 줄에 지도의 세로 크기 N과 가로 크기 M이 주어진다. (3 ≤ N, M ≤ 8)

둘째 줄부터 N개의 줄에 지도의 모양이 주어진다.
0은 빈 칸, 1은 벽, 2는 바이러스가 있는 위치이다. 2의 개수는 2보다 크거나 같고, 10보다 작거나 같은 자연수이다.

빈 칸의 개수는 3개 이상이다.

📌 출력

첫째 줄에 얻을 수 있는 안전 영역의 최대 크기를 출력한다.


📝 코드

import java.io.*;
import java.util.ArrayDeque;
import java.util.Deque;
import java.util.Queue;
import java.util.StringTokenizer;


public class Main {
    static int N, M;
    static int map[][];
    static boolean visited[][];
    static int[] dx = {0, 1, 0, -1}; //동 남 서 북 x
    static int[] dy = {1, 0, -1, 0}; //동 남 서 북 y
    static int maxSafeCnt = 0;

    static class Virus {
        int x;
        int y;

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


    public static void main(String[] args) throws IOException {

        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));

        StringTokenizer st = new StringTokenizer(br.readLine(), " ");

        N = Integer.parseInt(st.nextToken()); //세로
        M = Integer.parseInt(st.nextToken()); //가로

        visited = new boolean[N][M];
        map = new int[N][M];

        for (int i = 0; i < N; i++) {
            StringTokenizer stD = new StringTokenizer(br.readLine(), " ");
            for (int j = 0; j < M; j++) {
                map[i][j] = Integer.parseInt(stD.nextToken());
            }
        }

        dfs(0);

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

    static void dfs(int depth) {
        if (depth == 3) {
            bfs(); //벽이 완성 된 상태의 바이러스 안전지대 확인
            return;
        }

        for (int i = 0; i < N; i++) {
            for (int j = 0; j < M; j++) {
                if (map[i][j] == 0) { //벽을 세울 수 있는 빈 공간 이라면
                    map[i][j] = 1; //현재 위치에 벽을 세우고
                    dfs(depth + 1); //depth 1 추가 후 재귀
                    map[i][j] = 0; //현재 위치를 다시 빈공간으로 복귀
                    //이어서 반복분
                }
            }
        }
    }

    static void bfs() {
        int[][] virusMap = new int[N][M]; //virus Map
        //bfs 로 탐색하면서 내부 값을 변경 하기 때문에 임시 map 생성
        boolean[][] visited = new boolean[N][M]; //방문

        Queue<Virus> q = new ArrayDeque<>();

        for (int i = 0; i < N; i++) { //맵 복사
            for (int j = 0; j < M; j++) {
                virusMap[i][j] = map[i][j];
                if (map[i][j] == 2) { //virus 영역 이라면
                    q.offer(new Virus(i, j)); //큐에 추가
                    visited[i][j] = true;
                }
            }
        }

        //현재 까지 Queue 에 모든 virus 좌표 값이 저장 되어 있음

        while (!q.isEmpty()) {
            Virus virus = q.poll();
            int cx = virus.x; //현재 x좌표
            int cy = virus.y; //현재 y좌표

            for (int i = 0; i < 4; i++) {
                int nx = cx + dx[i]; //next x
                int ny = cy + dy[i]; //next y

                if (0 <= nx && nx < N && 0 <= ny && ny < M) {
                    if (visited[nx][ny] == false && virusMap[nx][ny] == 0) { //방문하지 않은 상태 , 빈공간 이라면
                        visited[nx][ny] = true; //다음 영역 visited true
                        virusMap[nx][ny] = 2; //다음 영역을 virus로 변경
                        q.offer(new Virus(nx, ny));
                    }
                }
            }
        }

        //virus 전부 변환 후
        //안전 영역 숫자 check
        int safeCnt = checkSafeArea(virusMap);

        maxSafeCnt = Math.max(safeCnt, maxSafeCnt);
    }

    static int checkSafeArea(int[][] virusMap) {
        int safeCnt = 0;

        for (int i = 0; i < N; i++) { //안전 영역 체크
            for (int j = 0; j < M; j++) {
                if (virusMap[i][j] == 0) {
                    safeCnt++;
                }
            }
        }

        return safeCnt;
    }

}

📗 정리

해당 코드는 모든 빈 공간에 벽을 세우면서 탐색 하기 때문에 많은 반복문 및 재귀를 필요로 한다.

따라서 코드를 짜면서도 시간 초과 걱정이 있었는데 다행히 주어진 시간을 초과 하지는 않았다.

벽을 세우는 dfs와 바이러스를 전염 시키는 bfs를 둘다 사용 했기에 코드 줄이 꽤 길다.

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

profile
wkd86591247@gmail.com

0개의 댓글