[백준 5214] 환승 - Java 문제풀이

문제 링크: https://www.acmicpc.net/problem/5214

 

문제

아주 먼 미래에 사람들이 가장 많이 사용하는 대중교통은 하이퍼튜브이다. 하이퍼튜브 하나는 역 K개를 서로 연결한다. 1번역에서 N번역으로 가는데 방문하는 최소 역의 수는 몇 개일까?

 

입력

첫째 줄에 역의 수 N과 한 하이퍼튜브가 서로 연결하는 역의 개수 K, 하이퍼튜브의 개수 M이 주어진다. (1 ≤ N ≤ 100,000, 1 ≤ K, M ≤ 1000)

다음 M개 줄에는 하이퍼튜브의 정보가 한 줄에 하나씩 주어진다. 총 K개 숫자가 주어지며, 이 숫자는 그 하이퍼튜브가 서로 연결하는 역의 번호이다. 

 

출력

첫째 줄에 1번역에서 N번역으로 가는데 방문하는 역의 개수의 최솟값을 출력한다. 만약, 갈 수 없다면 -1을 출력한다.

풀이

접근법

1. 모든 노드에 대해서 List로 갈 수 있는 모든 노드를 추가하는 것은 메모리가 너무 많이 낭비된다.
2. 그래서 하이퍼튜브에 대해서 하나의 자료구조로 두고, 이걸 전체로 관리하자
3. 어떤 노드에서 갈 수 있는 노드에 대해 그냥 하이퍼노드의 주소값을 추가하여 이걸 돌면서 사용하자
4. 근데 하나의 하이퍼노드를 여러번 도는 것은 비효율적이니까, 해당 하이퍼노드도 사용한 것인지를 체크하며 사용하자

 

코드

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

public class Main {
    // 2. 그래서 하이퍼튜브에 대해서 하나의 자료구조로 두고, 이걸 전체로 관리하자
    private static class SetNode {
        boolean visited; // 하이퍼튜브 방문 한건지
        Set<Integer> set; // set인데 list로 했어도 되었을 것 같다.
        public SetNode() {
            this.set = new HashSet<>();
        }
    }
    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());
        int K=Integer.parseInt(st.nextToken());
        int M=Integer.parseInt(st.nextToken());
        
        // 1. 모든 노드에 대해서 List로 갈 수 있는 모든 노드를 추가하는 것은 메모리가 너무 많이 낭비된다.
        List<SetNode>[] list = new List[N+1];
        for(int i=0;i<=N;i++) list[i] = new ArrayList<>();
        for(int i=0;i<M;i++) {
            st = new StringTokenizer(br.readLine());
            SetNode set = new SetNode();
            for(int j=0;j<K;j++) set.set.add(Integer.parseInt(st.nextToken()));
            // 3. 어떤 노드에서 갈 수 있는 노드에 대해 그냥 하이퍼노드의 주소값을 추가하여 이걸 돌면서 사용하자
            for(Integer num : set.set) list[num].add(set);
        }
        int[] visited = new int[N+1]; // 해당 노드를 방문한 것인지 체크
        Arrays.fill(visited, -1);
        Queue<Integer> queue = new LinkedList<>();
        queue.add(1);
        visited[1] = 1;
        outer: while(!queue.isEmpty()) {
            int now = queue.poll();
            for(SetNode nxtSet : list[now]) {
                // 4. 근데 하나의 하이퍼노드를 여러번 도는 것은 비효율적이니까, 해당 하이퍼노드도 사용한 것인지를 체크하며 사용하자
                if (nxtSet.visited) continue;
                nxtSet.visited = true;
                for(Integer nxt : nxtSet.set) { // 하이퍼 튜브 내 노드
                    if (visited[nxt] != -1) continue;
                    queue.add(nxt);
                    visited[nxt] = visited[now] + 1;
                    if (nxt == N) break outer;
                }
            }
        }
        System.out.println(visited[N]);
    }
}

 

20분 걸렸다 !