[Gold IV] Title: 특정한 최단 경로, Time: 616 ms, Memory: 70660 KB -BaekjoonHub

Author: CheChe903

백준/Gold/1504. 특정한 최단 경로/README.md

@@ -0,0 +1,30 @@
+# [Gold IV] 특정한 최단 경로 - 1504 
+
+[문제 링크](https://www.acmicpc.net/problem/1504) 
+
+### 성능 요약
+
+메모리: 70660 KB, 시간: 616 ms
+
+### 분류
+
+그래프 이론, 최단 경로, 데이크스트라
+
+### 제출 일자
+
+2025년 7월 1일 17:22:01
+
+### 문제 설명
+
+<p>방향성이 없는 그래프가 주어진다. 세준이는 1번 정점에서 N번 정점으로 최단 거리로 이동하려고 한다. 또한 세준이는 두 가지 조건을 만족하면서 이동하는 특정한 최단 경로를 구하고 싶은데, 그것은 바로 임의로 주어진 두 정점은 반드시 통과해야 한다는 것이다.</p>
+
+<p>세준이는 한번 이동했던 정점은 물론, 한번 이동했던 간선도 다시 이동할 수 있다. 하지만 반드시 최단 경로로 이동해야 한다는 사실에 주의하라. 1번 정점에서 N번 정점으로 이동할 때, 주어진 두 정점을 반드시 거치면서 최단 경로로 이동하는 프로그램을 작성하시오.</p>
+
+### 입력 
+
+ <p>첫째 줄에 정점의 개수 N과 간선의 개수 E가 주어진다. (2 ≤ N ≤ 800, 0 ≤ E ≤ 200,000) 둘째 줄부터 E개의 줄에 걸쳐서 세 개의 정수 a, b, c가 주어지는데, a번 정점에서 b번 정점까지 양방향 길이 존재하며, 그 거리가 c라는 뜻이다. (1 ≤ c ≤ 1,000) 다음 줄에는 반드시 거쳐야 하는 두 개의 서로 다른 정점 번호 v<sub>1</sub>과 v<sub>2</sub>가 주어진다. (v<sub>1</sub> ≠ v<sub>2</sub>, v<sub>1</sub> ≠ N, v<sub>2</sub> ≠ 1) 임의의 두 정점 u와 v사이에는 간선이 최대 1개 존재한다.</p>
+
+### 출력 
+
+ <p>첫째 줄에 두 개의 정점을 지나는 최단 경로의 길이를 출력한다. 그러한 경로가 없을 때에는 -1을 출력한다.</p>
+

백준/Gold/1504. 특정한 최단 경로/특정한 최단 경로.java

@@ -0,0 +1,98 @@
+import java.io.BufferedReader;
+import java.io.InputStreamReader;
+import java.io.IOException;
+import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.Comparator;
+import java.util.LinkedList;
+import java.util.PriorityQueue;
+import java.util.Queue;
+import java.util.StringTokenizer;
+
+public class Main {
+
+    static int N,E;
+    static ArrayList<int []> A[];
+    static int INF = Integer.MAX_VALUE;
+    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());
+
+        E = Integer.parseInt(st.nextToken());
+
+        A = new ArrayList[N+1];
+
+        for(int i =1; i<=N; i++) {
+            A[i] = new ArrayList<>();
+        }
+
+        for(int i =0; i<E; i++) {
+            st = new StringTokenizer(br.readLine());
+
+            int u = Integer.parseInt(st.nextToken());
+            int v = Integer.parseInt(st.nextToken());
+            int w = Integer.parseInt(st.nextToken());
+
+            A[u].add(new int[] {v,w});
+            A[v].add(new int[] {u,w});
+        }
+        st = new StringTokenizer(br.readLine());
+
+        int v1 = Integer.parseInt(st.nextToken());
+        int v2 = Integer.parseInt(st.nextToken());
+
+        long d1 = dijstra(1, v1);
+        long d2 = dijstra(v1, v2);
+        long d3 = dijstra(v2, N);
+        long routeA = (d1 >= INF || d2 >= INF || d3 >= INF) ? INF : d1 + d2 + d3;
+
+        // 경로 B: 1 → v2 → v1 → N
+        long e1 = dijstra(1, v2);
+        long e2 = dijstra(v2, v1);
+        long e3 = dijstra(v1, N);
+        long routeB = (e1 >= INF || e2 >= INF || e3 >= INF) ? INF : e1 + e2 + e3;
+
+        long answer = Math.min(routeA, routeB);
+        System.out.println(answer >= INF ? -1 : answer);
+    }
+
+    static long dijstra(int startNode, int endNode) {
+
+        PriorityQueue<long []> pq = new PriorityQueue<>(Comparator.comparingLong(a -> a[1]));
+
+        pq.add(new long[] {startNode, 0});
+
+        long diff[] = new long[N+1];
+
+        Arrays.fill(diff, Integer.MAX_VALUE);
+
+        diff[startNode] = 0;
+
+        while(!pq.isEmpty()) {
+            long now[] = pq.poll();
+
+            int currentNode = (int) now[0];
+            long currentValue = now[1];
+
+            if(currentValue > diff[currentNode]) {
+                continue;
+            }
+
+            for(int next[] : A[currentNode]) {
+                int nextNode = next[0];
+                int nextValue = next[1];
+
+                if(diff[nextNode] > currentValue + nextValue) {
+                    diff[nextNode] = currentValue + nextValue;
+                    pq.add(new long[] { nextNode, diff[nextNode]});
+                }
+            }
+        }
+
+        return diff[endNode];
+    }
+}