結果

問題 No.160 最短経路のうち辞書順最小
ユーザー jp_stejp_ste
提出日時 2016-06-06 22:26:39
言語 Java21
(openjdk 21)
結果
AC  
実行時間 113 ms / 5,000 ms
コード長 5,061 bytes
コンパイル時間 2,258 ms
コンパイル使用メモリ 79,408 KB
実行使用メモリ 52,124 KB
最終ジャッジ日時 2024-04-17 05:23:02
合計ジャッジ時間 5,219 ms
ジャッジサーバーID
(参考情報)
judge2 / judge4
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 53 ms
50,068 KB
testcase_01 AC 52 ms
50,304 KB
testcase_02 AC 52 ms
50,052 KB
testcase_03 AC 52 ms
50,356 KB
testcase_04 AC 85 ms
52,020 KB
testcase_05 AC 106 ms
52,124 KB
testcase_06 AC 113 ms
52,116 KB
testcase_07 AC 59 ms
50,536 KB
testcase_08 AC 64 ms
50,424 KB
testcase_09 AC 61 ms
50,112 KB
testcase_10 AC 66 ms
50,908 KB
testcase_11 AC 67 ms
50,696 KB
testcase_12 AC 66 ms
50,800 KB
testcase_13 AC 61 ms
50,396 KB
testcase_14 AC 62 ms
50,476 KB
testcase_15 AC 60 ms
50,028 KB
testcase_16 AC 76 ms
50,716 KB
testcase_17 AC 61 ms
50,592 KB
testcase_18 AC 63 ms
50,304 KB
testcase_19 AC 68 ms
50,936 KB
testcase_20 AC 78 ms
50,956 KB
testcase_21 AC 59 ms
50,356 KB
testcase_22 AC 59 ms
50,432 KB
testcase_23 AC 76 ms
50,652 KB
testcase_24 AC 69 ms
50,888 KB
testcase_25 AC 64 ms
50,280 KB
testcase_26 AC 61 ms
50,532 KB
testcase_27 AC 54 ms
50,364 KB
testcase_28 AC 100 ms
51,820 KB
testcase_29 AC 54 ms
50,216 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

import java.io.IOException;
import java.io.InputStream;
import java.io.PrintWriter;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.LinkedList;
import java.util.NoSuchElementException;

public class Main {
    static FastScanner scan = new FastScanner();
    static PrintWriter out = new PrintWriter(System.out);
    
    static int n, m, s, g;
    
    public static void main(String[] args) {
        n = scan.nextInt();
        m = scan.nextInt();
        s = scan.nextInt();
        g = scan.nextInt();
        
        Node[] nodes = new Node[n];
        long[] dp = new long[n];
        int[] path = new int[n];
        
        for(int i=0; i<n; i++) {
            nodes[i] = new Node(i);
            dp[i] = -1;
            path[i] = -1;
        }
        
        for(int i=0; i<m; i++) {
            int a = scan.nextInt();
            int b = scan.nextInt();
            int c = scan.nextInt();
            nodes[a].edges.add(new Edge(b, c));
            nodes[b].edges.add(new Edge(a, c));
        }
        
        dijkstra(nodes, dp, path, g);
        
        StringBuilder sb = new StringBuilder();
        sb.append(s);
        int next = path[s];
        while(next != -1) {
            sb.append(" ");
            sb.append(next);
            next = path[next];
        }
        System.out.println(sb.toString());
    }

    static void dijkstra(Node[] nodes, long[] dp, int[] path, int s) {

        Arrays.fill(dp, -1);
        dp[s] = 0;

        LinkedList<Node> q = new LinkedList<>();
        q.add(nodes[s]);

        while (!q.isEmpty()) {

            Node now = q.poll();

            for (int i = 0; i < now.edges.size(); i++) {
                int toId = now.edges.get(i).toId;
                long toCost = now.edges.get(i).toCost;

                if(dp[now.id] == -1) continue;
                
                long nextCost = dp[now.id] + toCost;
                if (dp[toId] < 0 || nextCost <= dp[toId]) {
                    
                    if(nextCost == dp[toId]) {
                        if(path[toId] > now.id) {
                            path[toId] = now.id;
                        }
                    } else {
                        dp[toId] = nextCost;
                        path[toId] = now.id;
                        q.add(nodes[toId]);
                    }
                }
            }
        }
    }

    static class Node {
        int id;
        ArrayList<Edge> edges = new ArrayList<>();

        Node(int id) {
            this.id = id;
        }
    }

    static class Edge {
        int toId;
        long toCost;

        Edge(int id, long cost) {
            toId = id;
            toCost = cost;
        }
    }
    
    static class FastScanner {
        private final InputStream in = System.in;
        private final byte[] buffer = new byte[1024];
        private int ptr = 0;
        private int buflen = 0;

        private boolean hasNextByte() {
            if (ptr < buflen) {
                return true;
            } else {
                ptr = 0;
                try {
                    buflen = in.read(buffer);
                } catch (IOException e) {
                    e.printStackTrace();
                }
                if (buflen <= 0) {
                    return false;
                }
            }
            return true;
        }

        private int readByte() {
            if (hasNextByte())
                return buffer[ptr++];
            else
                return -1;
        }

        private static boolean isPrintableChar(int c) {
            return 33 <= c && c <= 126;
        }

        private void skipUnprintable() {
            while (hasNextByte() && !isPrintableChar(buffer[ptr]))
                ptr++;
        }

        public boolean hasNext() {
            skipUnprintable();
            return hasNextByte();
        }

        public String next() {
            if (!hasNext())
                throw new NoSuchElementException();
            StringBuilder sb = new StringBuilder();
            int b = readByte();
            while (isPrintableChar(b)) {
                sb.appendCodePoint(b);
                b = readByte();
            }
            return sb.toString();
        }

        public int nextInt() {
            if (!hasNext())
                throw new NoSuchElementException();
            int n = 0;
            boolean minus = false;
            int b = readByte();
            if (b == '-') {
                minus = true;
                b = readByte();
            }
            if (b < '0' || '9' < b) {
                throw new NumberFormatException();
            }
            while (true) {
                if ('0' <= b && b <= '9') {
                    n *= 10;
                    n += b - '0';
                } else if (b == -1 || !isPrintableChar(b)) {
                    return minus ? -n : n;
                } else {
                    throw new NumberFormatException();
                }
                b = readByte();
            }
        }
    }
}
0