結果

問題 No.807 umg tours
ユーザー neko_the_shadowneko_the_shadow
提出日時 2019-05-16 17:33:05
言語 Java21
(openjdk 21)
結果
WA  
実行時間 -
コード長 3,233 bytes
コンパイル時間 2,823 ms
コンパイル使用メモリ 84,484 KB
実行使用メモリ 129,848 KB
最終ジャッジ日時 2024-09-17 05:37:38
合計ジャッジ時間 50,089 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 181 ms
59,988 KB
testcase_01 AC 185 ms
54,720 KB
testcase_02 AC 210 ms
54,344 KB
testcase_03 AC 211 ms
54,184 KB
testcase_04 AC 186 ms
54,372 KB
testcase_05 AC 186 ms
54,264 KB
testcase_06 AC 221 ms
54,672 KB
testcase_07 AC 195 ms
54,476 KB
testcase_08 AC 152 ms
54,412 KB
testcase_09 AC 162 ms
54,296 KB
testcase_10 AC 166 ms
54,788 KB
testcase_11 AC 2,841 ms
114,644 KB
testcase_12 AC 2,862 ms
114,948 KB
testcase_13 AC 3,130 ms
125,684 KB
testcase_14 AC 2,048 ms
82,712 KB
testcase_15 AC 1,708 ms
74,992 KB
testcase_16 AC 3,074 ms
117,200 KB
testcase_17 AC 3,744 ms
129,848 KB
testcase_18 AC 3,819 ms
129,340 KB
testcase_19 AC 3,624 ms
128,752 KB
testcase_20 WA -
testcase_21 WA -
testcase_22 WA -
testcase_23 WA -
testcase_24 TLE -
testcase_25 -- -
権限があれば一括ダウンロードができます

ソースコード

diff # 

import java.util.Arrays;
import java.util.Comparator;
import java.util.HashMap;
import java.util.Map;
import java.util.Map.Entry;
import java.util.PriorityQueue;
import java.util.Scanner;
import java.util.function.Function;
import java.util.stream.Collectors;
import java.util.stream.IntStream;

public class Main {
    public static void main(String[] args) {
        Scanner stdin = new Scanner(System.in);
        
        String[] first = stdin.nextLine().split(" ");
        int n = Integer.parseInt(first[0]);
        int m = Integer.parseInt(first[1]);
        
        Map<Integer, Map<Integer, Integer>> costs 
            = IntStream.range(0, n).boxed().collect(Collectors.toMap(Function.identity(), unused -> new HashMap<>()));
        
        for (int i = 0; i < m; i++) {
            String[] line = stdin.nextLine().split(" ");
            int a = Integer.parseInt(line[0]) - 1;
            int b = Integer.parseInt(line[1]) - 1;
            int c = Integer.parseInt(line[2]);
            
            costs.get(a).put(b, c);
            costs.get(b).put(a, c);
        }
        
        int[] d1 = new int[n];
        PriorityQueue<Integer> q1 = new PriorityQueue<>(Comparator.comparing(i -> d1[i]));
        
        Arrays.fill(d1, Integer.MAX_VALUE);
        d1[0] = 0;
        q1.add(0);
        
        while (!q1.isEmpty()) {
            int u = q1.poll();
            for (Entry<Integer, Integer> e : costs.get(u).entrySet()) {
                int v = e.getKey();
                if (d1[u] + e.getValue() < d1[v]) {
                    d1[v] = d1[u] + e.getValue();
                    q1.add(v);
                }
            }
        }
        
        
        int[] d2 = new int[n * 2];
        PriorityQueue<Integer> q2 = new PriorityQueue<>(Comparator.comparing(i -> d2[i]));
        
        Arrays.fill(d2, Integer.MAX_VALUE);
        d2[n * 0 + 0] = 0;
        d2[n * 1 + 0] = 0;
        q2.add(n * 0 + 0);
        q2.add(n * 1 + 0);
        
        while (!q2.isEmpty()) {
            int u = q2.poll();
            int x = u % n;
            boolean skipped = (u / n == 1);
            if (skipped) {
                for (Entry<Integer, Integer> e : costs.get(x).entrySet()) {
                    int y = e.getKey();
                    int v = n * 1 + y;
                    if (d2[u] + e.getValue() < d2[v]) {
                        d2[v] = d2[u] + e.getValue();
                        q2.add(v);
                    }
                }
            } else {
                for (Entry<Integer, Integer> e : costs.get(x).entrySet()) {
                    int y1 = e.getKey();
                    int v1 = n * 0 + y1;
                    if (d2[u] + e.getValue() < d2[v1]) {
                        d2[v1] = d2[u] + e.getValue();
                        q2.add(v1);
                    }
                    
                    int y2 = e.getKey();
                    int v2 = n * 1 + y2;
                    if (d2[u] < d2[v2]) {
                        d2[v2] = d2[u];
                        q2.add(v2);
                    }
                }
            }
        }
        
        for (int i = 0; i < n; i++) {
            System.out.println(d1[i] + d2[n * 1 + i]);
        }
    }
}
0