[백준] 2146 다리 만들기.Java

9999·2023년 5월 12일
0

BOJ

목록 보기
20/128

문제

여러 섬으로 이루어진 나라가 있다. 이 나라의 대통령은 섬을 잇는 다리를 만들겠다는 공약으로 인기몰이를 해 당선될 수 있었다. 하지만 막상 대통령에 취임하자, 다리를 놓는다는 것이 아깝다는 생각을 하게 되었다. 그래서 그는, 생색내는 식으로 한 섬과 다른 섬을 잇는 다리 하나만을 만들기로 하였고, 그 또한 다리를 가장 짧게 하여 돈을 아끼려 하였다.

이 나라는 N×N크기의 이차원 평면상에 존재한다. 이 나라는 여러 섬으로 이루어져 있으며, 섬이란 동서남북으로 육지가 붙어있는 덩어리를 말한다. 다음은 세 개의 섬으로 이루어진 나라의 지도이다.
위의 그림에서 색이 있는 부분이 육지이고, 색이 없는 부분이 바다이다. 이 바다에 가장 짧은 다리를 놓아 두 대륙을 연결하고자 한다. 가장 짧은 다리란, 다리가 격자에서 차지하는 칸의 수가 가장 작은 다리를 말한다. 다음 그림에서 두 대륙을 연결하는 다리를 볼 수 있다.

물론 위의 방법 외에도 다리를 놓는 방법이 여러 가지 있으나, 위의 경우가 놓는 다리의 길이가 3으로 가장 짧다(물론 길이가 3인 다른 다리를 놓을 수 있는 방법도 몇 가지 있다).

지도가 주어질 때, 가장 짧은 다리 하나를 놓아 두 대륙을 연결하는 방법을 찾으시오.

입력

첫 줄에는 지도의 크기 N(100이하의 자연수)가 주어진다. 그 다음 N줄에는 N개의 숫자가 빈칸을 사이에 두고 주어지며, 0은 바다, 1은 육지를 나타낸다. 항상 두 개 이상의 섬이 있는 데이터만 입력으로 주어진다.

출력

첫째 줄에 가장 짧은 다리의 길이를 출력한다.

import java.io.*;
import java.util.*;
public class Main  {
    static int N, count, min;
    static int[][] map;
    static boolean[][] visited, bridgeCk;
    static int[] dx = { 1, -1, 0, 0 };
    static int[] dy = { 0, 0, 1, -1 };
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        N = Integer.parseInt(br.readLine());
        map = new int[N][N];
        for (int i = 0; i < N; i++) {
            String[] str = br.readLine().split(" ");
            for (int j = 0; j < N; j++) {
                map[i][j] = Integer.parseInt(str[j]);
            }
        }
        visited = new boolean[N][N];

        // 대륙 분류.
        count = 1;
        for (int i = 0; i < N; i++) {
            for (int j = 0; j < N; j++) {
                if (!visited[i][j] && map[i][j] != 0) {
                    mapClassify(i, j);
                }
            }
        }

        min = Integer.MAX_VALUE;
        bridgeCk = new boolean[N][N];
        // 테두리 좌표 수집후 최소거리 구하기.
        for (int i = 0; i < N; i++) {
            for (int j = 0; j < N; j++) {
                if (map[i][j] != 0 && !bridgeCk[i][j]) {
                    BFS(i, j);
                    bridgeCk = new boolean[N][N];
                }
            }
        }

        System.out.println(min == Integer.MAX_VALUE ? 0 : min);
    }
    public static void BFS(int x, int y) {
        int target = map[x][y];
        Queue<Node> q = new LinkedList<>();
        q.add(new Node(x, y));
        bridgeCk[x][y] = true;
        while (!q.isEmpty()) {
            Node n = q.poll();
            for (int i = 0; i < 4; i++) {
                int nx = n.x + dx[i];
                int ny = n.y + dy[i];
                if (nx < 0 || ny < 0 || nx >= N || ny >= N)
                    continue;
                if (bridgeCk[nx][ny])
                    continue;
                if (map[nx][ny] == target)
                    continue;
                if (map[nx][ny] == 0) {
                    q.add(new Node(nx, ny, n.dist + 1));
                    bridgeCk[nx][ny] = true;
                } else min = Math.min(min, n.dist);
            }
        }
    }
    public static void mapClassify(int x, int y) { // 대륙 분류.
        Queue<Node> q = new LinkedList<>();
        q.add(new Node(x, y));
        visited[x][y] = true;
        map[x][y] = count;
        while(!q.isEmpty()) {
            Node n = q.poll();
            for (int i = 0; i < 4; i++) {
                int nx = n.x + dx[i];
                int ny = n.y + dy[i];
                if (nx < 0 || ny < 0 || nx >= N || ny >= N)
                    continue;
                if (map[nx][ny] == 0)
                    continue;
                if (!visited[nx][ny]) {
                    visited[nx][ny] = true;
                    map[nx][ny] = count;
                    q.add(new Node(nx, ny));
                }
            }
        }
        count++;
    }

}
class Node {
    int x, y, dist;
    Node (int x, int y) {
        this.x = x;
        this.y = y;
    }
    Node (int x, int y, int dist) {
        this.x = x;
        this.y = y;
        this.dist = dist;
    }
}
  • 원래는 이 순서로 하려고 했다.
    1. 대륙 분류
    2. 테두리 좌표만 뽑아서 리스트에 저장.
    3. 리스를 반복문으로 돌면서 같은 좌표값끼리 더해서 -1하면 최소거리가 나온다.
  • 2번까지 구현했지만 방문처리가 비효율적이라고 생각해서 중단.
  • 그리고 굳이 리스트를 따로 안뽑고 그냥 같은 대륙을 제외하고 거리 최솟값만 구하면 된다.

0개의 댓글