Algorithm/백준

백준 [17086] 아기 상어 2 (자바)

youngjae-kim 2024. 6. 7. 10:55
728x90
반응형

문제

풀이

  1. 아기 상어가 있지 않은 모든 칸으로 부터 아기 상어까지 안전 거리를 구한다.
  2. 방향 배열 대각선 포함 8 방향을 bfs 탐색하며 진행
  3. 제일 먼저 아기 상어를 만나는 거리가 안전거리이므로 아기상어를 만나면 탐색 종료
  4. 안전 거리 최대 값 갱신

정답코드

package com.baekjoon.p17086;

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

public class Main {
    static int N, M, max;
    static int[][] map;

    static int[] dr = { 1, 1, 0, -1, -1, -1, 0, 1 };
    static int[] dc = { 0, -1, -1, -1, 0, 1, 1, 1 };

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

        StringTokenizer st = new StringTokenizer(br.readLine());
        N = Integer.parseInt(st.nextToken());
        M = Integer.parseInt(st.nextToken());

        map = new int[N][M];

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

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

        for (int i = 0; i < N; i++) {
            for (int j = 0; j < M; j++) {
                if (map[i][j] == 0) {
                    bfs(i, j);
                }
            }
        }

        System.out.println(max);
    }

    static void bfs(int sr, int sc) {
        Queue<int[]> q = new ArrayDeque<>();
        boolean[][] v = new boolean[N][M];

        v[sr][sc] = true;
        q.add(new int[] { sr, sc, 0 });
        int dist = 0;

        while (!q.isEmpty()) {
            int[] now = q.poll();

            if (map[now[0]][now[1]] == 1) {
                if (dist < now[2]) {
                    dist = now[2];
                    break;
                }
            }

            for (int i = 0; i < 8; i++) {
                int nr = now[0] + dr[i];
                int nc = now[1] + dc[i];

                if (nr < 0 || nr >= N || nc < 0 || nc >= M || v[nr][nc]) {
                    continue;
                }

                v[nr][nc] = true;
                q.add(new int[] { nr, nc, now[2] + 1 });
            }
        }

        if (max < dist) {
            max = dist;
        }
    }
}
728x90
반응형