본문 바로가기

코딩 테스트/BAEKJOON

[Java] 6118 숨바꼭질

문제

문제
재서기는 수혀니와 교외 농장에서 숨바꼭질을 하고 있다.
농장에는 헛간이 많이 널려있고 재서기는 그 중에 하나에 숨어야 한다.
헛간의 개수는 N(2 <= N <= 20,000)개이며, 1 부터 샌다고 하자.  

재서기는 수혀니가 1번 헛간부터 찾을 것을 알고 있다. 

모든 헛간은 M(1<= M <= 50,000)개의 양방향 길로 이어져 있고,
그 양 끝을 A_i 와 B_i(1<= A_i <= N; 1 <= B_i <= N; A_i != B_i)로 나타낸다. 

또한 어떤 헛간에서 다른 헛간으로는 언제나 도달 가능하다고 생각해도 좋다. 

재서기는 발냄새가 지독하기 때문에 최대한 냄새가 안나게 숨을 장소를 찾고자 한다. 

냄새는 1번 헛간에서의 거리(여기서 거리라 함은 지나야 하는 길의 최소 개수이다)가 
멀어질수록 감소한다고 한다. 

재서기의 발냄새를 최대한 숨길 수 있는 헛간을 찾을 수 있게 도와주자!

입력
첫 번째 줄에는 N과 M이 공백을 사이에 두고 주어진다.

이후 M줄에 걸쳐서 A_i와 B_i가 공백을 사이에 두고 주어진다.

 

출력
출력은 한줄로 이루어지며, 세 개의 값을 공백으로 구분지어 출력해야한다. 

첫 번째는 숨어야 하는 헛간 번호를(만약 거리가 같은 헛간이 여러개면 가장 작은 헛간 번호를 출력한다)
, 두 번째는 그 헛간까지의 거리를, 
세 번째는 그 헛간과 같은 거리를 갖는 헛간의 개수를 출력해야한다.

문제 풀이

그래프를 이용한 BFS 문제.

자바 코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.*;

public class N6118 {
    private static boolean[] checked;
    private static List<Integer>[] road;
    private static int distance, destination, cnt;
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine(), " ");

        int n = Integer.parseInt(st.nextToken()), m = Integer.parseInt(st.nextToken());
        checked = new boolean[n + 1];
        road = new ArrayList[n + 1];
        for(int i = 0; i < n + 1; i++)
            road[i] = new ArrayList<>();

        for(int i = 0; i < m; i++){
            st = new StringTokenizer(br.readLine(), " ");
            int a_i = Integer.parseInt(st.nextToken());
            int b_i = Integer.parseInt(st.nextToken());
            road[a_i].add(b_i);
            road[b_i].add(a_i);
        }

        bfs();
        System.out.println(destination + " " + distance +  " " + cnt);
    }

    private static void bfs() {
        Queue<int []> q = new LinkedList<>();
        checked[1] = true;
        // 현재 노드 값, 거리 값
        q.offer(new int[]{1,0});

        while(!q.isEmpty()){
            int arr[] = q.poll();
            int now = arr[0];
            int next = arr[1];

            if(next > distance){
                distance = next;
                destination = now;
                cnt = 1;
            }
            else if(next == distance){
                if(destination > now) destination = now;
                cnt++;
            }

            for(int i = 0; i < road[now].size(); i++){
                int next_next = road[now].get(i);
                if(!checked[next_next]){
                    checked[next_next] = true;
                    q.offer(new int[]{next_next,next+1});
                }
            }
        }
    }
}

 

'코딩 테스트 > BAEKJOON' 카테고리의 다른 글

[Java] 12851 숨바꼭질 2  (0) 2024.04.30
[Java] 14940 쉬운 최단거리  (0) 2024.04.30
[Java] 2644 촌수계산  (0) 2024.04.29
[Java] 16973 직사각형 탈출  (0) 2024.04.27
[Java] 2252 줄 세우기  (1) 2024.04.25