[Graph] baekjoon-5719

Author: Seogeurim

README.md

@@ -193,6 +193,7 @@
 | 05  |     | [Baekjoon-11404 플로이드](./src/Graph/P11404)        |   Floyd-Warshall   |
 | 06  |     | [Baekjoon-1854 K번째 최단경로 찾기](./src/Graph/P1854)   |   Dijkstra   |
 | 07  |     | [Baekjoon-3860 할로윈 묘지](./src/Graph/P3860)        |   Bellman-Ford   |
+| 06  | ⭐️  | [Baekjoon-5719 거의 최단 경로](./src/Graph/P5719)      |   Dijkstra   |
 
 ### Union-Find
 

src/Graph/P5719/Main.java

@@ -0,0 +1,96 @@
+package Graph.P5719;
+
+import java.io.*;
+import java.util.*;
+
+public class Main {
+
+    static int N, M, S, D, U, V, P;
+    static ArrayList<Node>[] graph, reverse;
+    static int[] dist;
+
+    public static void main(String[] args) throws Exception {
+        System.setIn(new FileInputStream("src/Graph/P5719/input.txt"));
+        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
+        StringTokenizer st = new StringTokenizer(br.readLine());
+
+        while ((N = stoi(st.nextToken())) != 0 && (M = stoi(st.nextToken())) != 0) {
+            st = new StringTokenizer(br.readLine());
+            S = stoi(st.nextToken()); D = stoi(st.nextToken());
+
+            graph = new ArrayList[N];
+            reverse = new ArrayList[N];
+            for (int i = 0; i < N; i++) {
+                graph[i] = new ArrayList<>();
+                reverse[i] = new ArrayList<>();
+            }
+
+            while (M-- > 0) {
+                st = new StringTokenizer(br.readLine());
+                U = stoi(st.nextToken()); V = stoi(st.nextToken()); P = stoi(st.nextToken());
+                graph[U].add(new Node(V, P));
+                reverse[V].add(new Node(U, P));
+            }
+
+            dijkstra();
+            delPath();
+            dijkstra();
+            System.out.println(dist[D]);
+
+            st = new StringTokenizer(br.readLine());
+        }
+    }
+
+    static void dijkstra() {
+        dist = new int[N];
+        Arrays.fill(dist, -1);
+
+        PriorityQueue<Node> pq = new PriorityQueue<>(((o1, o2) -> o1.dist - o2.dist));
+        pq.offer(new Node(S, 0));
+        while (!pq.isEmpty()) {
+            Node cur = pq.poll();
+            if (dist[cur.idx] != -1) continue;
+
+            dist[cur.idx] = cur.dist;
+            for (Node next : graph[cur.idx]) {
+                if (dist[next.idx] == -1)
+                    pq.offer(new Node(next.idx, dist[cur.idx] + next.dist));
+            }
+        }
+    }
+
+    static void delPath() {
+        Queue<Integer> q = new LinkedList<>();
+        boolean[] visited = new boolean[N];
+        q.offer(D); visited[D] = true;
+        while (!q.isEmpty()) {
+            int cur = q.poll();
+            for (Node prev : reverse[cur]) {
+                if (dist[prev.idx] + prev.dist == dist[cur]) {
+                    graph[prev.idx].remove(new Node(cur, prev.dist));
+                    if (!visited[prev.idx]) {
+                        visited[prev.idx] = true;
+                        q.offer(prev.idx);
+                    }
+                }
+            }
+        }
+    }
+
+    static int stoi(String s) { return Integer.parseInt(s); }
+
+    static class Node {
+        int idx, dist;
+
+        public Node(int idx, int dist) {
+            this.idx = idx;
+            this.dist = dist;
+        }
+
+        @Override
+        public boolean equals(Object o) {
+            Node node = (Node) o;
+            return idx == node.idx && dist == node.dist;
+        }
+    }
+}

src/Graph/P5719/README.md

@@ -0,0 +1,19 @@
+## [baekjoon-5719] 거의 최단 경로
+
+![image](https://user-images.githubusercontent.com/22045163/107655671-00a46e80-6cc7-11eb-83e9-e2834b9a87be.png)
+![image](https://user-images.githubusercontent.com/22045163/107656046-19ad1f80-6cc7-11eb-80db-179aaa45aefc.png)
+
+### 풀이 과정
+
+다익스트라 응용 문제이다. 많이 어려웠다. ㅠㅠ
+
+처음에는 [K번째 최단경로 찾기](../P1854) 문제와 같은 방식으로 2번째 최단 경로를 찾으면 되는줄 알았으나, 이 문제는 그렇게 푸는 것이 아니라 
+**최적의 최단 경로를 찾은 후 그 최단 경로를 지우고, 다시 최단 경로를 찾아야** 올바른 정답을 도출할 수 있었다. 
+즉 다익스트라 -> 최단 경로 지우기 -> 다익스트라 순서로 풀이해야 하는 것이다.
+
+최단 경로를 지우는 방법은 마지막 노드에서 역순으로 거쳐온 노드를 찾아 해당 정보를 지우면 되고, bfs를 통해 해결할 수 있다. 
+이 때 방문했던 노드에 대해서는 방문 처리를 해주어야 bfs 메모리 초과 이슈를 피할 수 있다.
+
+![image](https://user-images.githubusercontent.com/22045163/107656230-4bbe8180-6cc7-11eb-910a-74af12747b03.png)
+
+![image](https://user-images.githubusercontent.com/22045163/107656082-23cf1e00-6cc7-11eb-84fd-3cf205abc5a8.png)

src/Graph/P5719/input.txt

@@ -0,0 +1,30 @@
+7 9
+0 6
+0 1 1
+0 2 1
+0 3 2
+0 4 3
+1 5 2
+2 6 4
+3 6 2
+4 6 4
+5 6 1
+4 6
+0 2
+0 1 1
+1 2 1
+1 3 1
+3 2 1
+2 0 3
+3 0 2
+6 8
+0 1
+0 1 1
+0 2 2
+0 3 3
+2 5 3
+3 4 2
+4 1 1
+5 1 1
+3 0 1
+0 0