[Algo] 백준 17396 백도어 JAVA

문제 바로가기 : https://www.acmicpc.net/problem/17396

메모리 : 135764KB

시간 : 904ms

언어 : Java 11

---

코드

import java.beans.beancontext.BeanContext;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.PriorityQueue;
import java.util.StringTokenizer;

public class Main {
    static class Node {
        int idx;
        long cost;

        Node(int idx, long cost) {
            this.idx = idx;
            this.cost = cost;
        }
    }

    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 M = Integer.parseInt(st.nextToken());
        st = new StringTokenizer(br.readLine());
        boolean[] view = new boolean[N];
        for (int i = 0; i < N; i++) {
            int t = Integer.parseInt(st.nextToken());
            if (t == 0) {
                view[i] = false;
            } else {
                view[i] = true;
            }
            // 넥서스는 시야 보여도 갈 수 있다.
            if (i == N - 1) {
                view[i] = false;
            }
        }

        // 인접리스트 생성
        ArrayList<ArrayList<Node>> list = new ArrayList<>();
        for (int i = 0; i < N; i++) {
            list.add(new ArrayList<>());
        }
        for (int i = 0; i < M; i++) {
            st = new StringTokenizer(br.readLine());
            int x = Integer.parseInt(st.nextToken());
            int y = Integer.parseInt(st.nextToken());
            int c = Integer.parseInt(st.nextToken());
            // 와드가 없으면 인접리스트에 저장
            if (!view[x] && !view[y]) {
                list.get(x).add(new Node(y, c));
                list.get(y).add(new Node(x, c));
            }
        }
        // 최단거리 dp배열 생성
        long[] dist = new long[N];
        for (int i = 0; i < N; i++) {
            dist[i] = Long.MAX_VALUE;
        }
        dist[0] = 0;
        // 적은 가격으로 정렬해서 작은 것부터 뽑기
        PriorityQueue<Node> pq = new PriorityQueue<>((o1, o2) -> Long.compare(o1.cost, o2.cost));
        // 맨 처음은 0번 노드 값 0
        pq.add(new Node(0, 0));

        while (!pq.isEmpty()) {
            // poll 노드는 항상 최소
            Node n = pq.poll();
            if (n.idx == N - 1) { // 종점까지 나왔으면 그만
                break;
            }
            // 지금 node 비용이 dist비용보다 크면 볼 필요 없다.
            if (dist[n.idx] < n.cost) {
                continue;
            }
            // 지금 비용이 더 작으면 주변 노드들을 갱신해줘야한다.
            for (Node next : list.get(n.idx)) {

                // 현재 dp되어있는 값 > 현재이동값+다음이동값 즉 새 길로 가는 것이 더 가까울 때 갱신한다.

                if (dist[next.idx] > next.cost + n.cost) {
                    dist[next.idx] = next.cost + n.cost;
                    // 갱신된 경우에 pq에 넣는다 = 출발지에서 갈 수 있는 정점이면 pq에 넣어서 최단거리 계산해야한다.
                    pq.add(new Node(next.idx, dist[next.idx]));
                }
            }
        }
        if (dist[N - 1] == Long.MAX_VALUE) {
            System.out.println(-1);
        } else {
            System.out.println(dist[N - 1]);
        }
    }
}

풀이

pq를 사용한 다익스트라.
pq를 사용한 버전은 bfs같은 느낌이 났다. 그래서 그리디스러운것인가보다!