글 읽기 - 다익스트라 문제 질문입니다

baepoce 2년 전 0

최단경로 먼저 구하고 가까워지는 경로 구하는 방법으로 시도했는데 실패했습니다.

어디부분이 틀렸는지 잘 모르겠습니다..

import static java.lang.Integer.parseInt;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.PriorityQueue;
import java.util.StringTokenizer;

public class Main {

    static int n, m, ans;
    static int MAX = 99999999;
    static int[] dist;
    static boolean[] visit;
    static List<Node>[] nodes;

    static class Node {

        int v, d;

        Node(int v, int d) {
            this.v = v;
            this.d = d;
        }
    }

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());
        n = parseInt(st.nextToken());
        m = parseInt(st.nextToken());
        dist = new int[n + 1];
        visit = new boolean[n + 1];
        Arrays.fill(dist, MAX);
        nodes = new ArrayList[n + 1];
        PriorityQueue<Integer> pq = new PriorityQueue<>((o1, o2) -> dist[o1] - dist[o2]);
        for (int i = 0; i <= n; i++) {
            nodes[i] = new ArrayList<>();
        }
        for (int i = 0; i < m; i++) {
            st = new StringTokenizer(br.readLine());
            int a = parseInt(st.nextToken());
            int b = parseInt(st.nextToken());
            int c = parseInt(st.nextToken());
            nodes[a].add(new Node(b, c));
            nodes[b].add(new Node(a, c));
            if (a == 1) {
                dist[b] = c;
            }
            if (b == 1) {
                dist[a] = c;
            }
        }
        pq.add(1);
        dist[1] = 0;
        while (!pq.isEmpty()) {
            int now = pq.poll();
            if (visit[now]) {
                continue;
            }
            visit[now] = true;
            for (Node node : nodes[now]) {
                int next = node.v;
                int distance = node.d;
                if (dist[next] >= dist[now] + distance) {
                    dist[next] = dist[now] + distance;
                    pq.add(next);
                }
            }
        }
        visit = new boolean[n + 1];
        dfs(1);
        System.out.println(ans);
    }

    private static void dfs(int now) {
        visit[now] = true;
        for (Node node : nodes[now]) {
            int next = node.v;
            if (next == 2) {
                if (node.d < dist[2]) {
                    ans++;
                }
            }
            if (!visit[next]) {
                dfs(next);
            }
        }
    }
}

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

다익스트라 문제 질문입니다

baepoce 2년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말