[BOJ] 1916 최소비용 구하기 - JAVA

1. 문제

1916번: 최소비용 구하기

N개의 도시가 있다. 그리고 한 도시에서 출발하여 다른 도시에 도착하는 M개의 버스가 있다. 우리는 A번째 도시에서 B번째 도시까지 가는데 드는 버스 비용을 최소화 시키려고 한다. A번째 도시에서 B번째 도시까지 가는데 드는 최소비용을 출력하여라. 도시의 번호는 1부터 N까지이다.

---

2. 풀이

기본적인 다익스트라 풀이다. 우선순위 큐를 이용했다.

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

public class BOJ_1916_최소비용구하기 {
   public static void main(String[] args) throws IOException {
      BufferedReader in = new BufferedReader(new InputStreamReader(System.in));
      int N = Integer.parseInt(in.readLine());
      int M = Integer.parseInt(in.readLine());
      int[][] fee = new int[N + 1][N + 1];
      int INF = 100000001;

      for (int i = 1; i <= N; i++) {
         Arrays.fill(fee[i], INF);
      }

      StringTokenizer st;
      for (int i = 0; i < M; i++) {
         st = new StringTokenizer(in.readLine());
         int start = Integer.parseInt(st.nextToken());
         int end = Integer.parseInt(st.nextToken());
         int cost = Integer.parseInt(st.nextToken());
         fee[start][end] = Math.min(fee[start][end],cost);
      }

      st = new StringTokenizer(in.readLine());
      int A = Integer.parseInt(st.nextToken());
      int B = Integer.parseInt(st.nextToken());

      int[] dist = new int[N + 1];
      Arrays.fill(dist, INF);
      dist[A] = 0;

      boolean[] visited = new boolean[N + 1];
      PriorityQueue<int[]> pq = new PriorityQueue<>((o1, o2) -> o1[1] - o2[1]);
      pq.offer(new int[]{A, dist[A]});


      int[] now;
      while (!pq.isEmpty()) {
         now = pq.poll();
         if (visited[now[0]]) continue;
         visited[now[0]] = true;
         dist[now[0]] = now[1];
         if (now[0] == B) {
            break;
         }

         for (int i = 1; i <= N; i++) {
            if (visited[i] || fee[now[0]][i] < 0) continue;
            if (dist[i] > now[1] + fee[now[0]][i]) {
               dist[i] = now[1]+fee[now[0]][i];
               pq.offer(new int[]{i, dist[i]});
            }
         }
      }
      System.out.println(dist[B]);

   }
}

---

3. 결과

ㅋㅋㅋ 한 도시에서 다른 도시까지 가는 버스가 여러 대인줄 모르고 그냥 저장했다가 2퍼만에 계속 틀리는 걸 반복했다..

겨우 알아냈다.

역시 문제를 똑바로 읽어야 한다.