[백준] 11779. 최소비용 구하기 2

문제 링크

[백준] 11779. 최소비용 구하기 2

풀이 과정

다익스트라에 최소 비용 경로까지 출력해야하는 문제입니다. parent 이름의 부모 노드 번호를 저장하는 배열을 이용해, 비용이 갱신될때마다 해당 정점의 부모 또한 갱신해주었습니다. 마지막으로 이를 역추적해서 경로를 출력할 수 있었습니다.

코드

import java.util.*;

public class Main {
    static int N, M, start, end;
    static final int MAX = 987654321;
    static ArrayList<Edge>[] adj;
    static int[] dist;
    static int[] parent;
    static boolean[] visit;

    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        N = sc.nextInt();
        M = sc.nextInt();
        adj = new ArrayList[N + 1];
        dist = new int[N + 1];
        parent = new int[N + 1];
        visit = new boolean[N + 1];

        for (int i = 1; i <= N; i++) adj[i] = new ArrayList<>();
        for (int i = 1; i <= N; i++) dist[i] = MAX;
        for (int i = 0; i < M; i++) adj[sc.nextInt()].add(new Edge(sc.nextInt(), sc.nextInt()));

        start = sc.nextInt();
        end = sc.nextInt();

        dijkstra();

        int x = end;
        Stack<Integer> stack = new Stack<>();

        while (x != parent[x]) {
            stack.push(x);
            x = parent[x];
        }

        System.out.println(dist[end]);
        System.out.println(stack.size());
        while (!stack.isEmpty())
            System.out.print(stack.pop() + " ");
    }

    static void dijkstra() {
        PriorityQueue<Edge> pq = new PriorityQueue<>();
        dist[start] = 0;
        pq.add(new Edge(start, 0));

        while (!pq.isEmpty()) {
            Edge now = pq.poll();

            for (Edge next : adj[now.vertex]) {
                if (dist[next.vertex] > dist[now.vertex] + next.cost) {
                    dist[next.vertex] = dist[now.vertex] + next.cost;
                    pq.add(new Edge(next.vertex, dist[next.vertex]));

                    parent[next.vertex] = now.vertex;
                }
            }
        }
    }

    static class Edge implements Comparable<Edge> {
        int vertex, cost;

        public Edge(int vertex, int cost) {
            this.vertex = vertex;
            this.cost = cost;
        }

        @Override
        public int compareTo(Edge o) {
            return this.cost - o.cost;
        }
    }
}