結果

問題 No.807 umg tours
ユーザー neko_the_shadow
提出日時 2019-05-16 17:46:01
言語 Java
(openjdk 23)
結果
WA  
実行時間 -
コード長 2,793 bytes
コンパイル時間 2,432 ms
コンパイル使用メモリ 95,484 KB
実行使用メモリ 765,864 KB
最終ジャッジ日時 2024-09-17 05:38:20
合計ジャッジ時間 21,928 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 5 WA * 6 MLE * 15
権限があれば一括ダウンロードができます

ソースコード

diff # 

import java.util.Arrays;
import java.util.Comparator;
import java.util.PriorityQueue;
import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Scanner stdin = new Scanner(System.in);
        int n = Integer.parseInt(stdin.next());
        int m = Integer.parseInt(stdin.next());
        
        long[][] costs = new long[n][n];
        for (long[] row : costs) {
            Arrays.fill(row, -1);
        }
        
        for (int i = 0; i < m; i++) {
            int a = Integer.parseInt(stdin.next()) - 1;
            int b = Integer.parseInt(stdin.next()) - 1;
            long c = Long.parseLong(stdin.next());
            
            costs[a][b] = c;
            costs[b][a] = c;
        }

        long[] d1 = new long[n];
        PriorityQueue<Integer> q1 = new PriorityQueue<>(Comparator.comparing(i -> d1[i]));
        
        Arrays.fill(d1, Long.MAX_VALUE);
        d1[0] = 0;
        q1.add(0);
        
        while (!q1.isEmpty()) {
            int u = q1.poll();
            for (int v = 0; v < n; v++) {
                if (costs[u][v] != -1 && d1[u] + costs[u][v] < d1[v]) {
                    d1[v] = d1[u] + costs[u][v];
                    q1.add(v);
                }
            }
        }
        
        
        long[] d2 = new long[n * 2];
        PriorityQueue<Integer> q2 = new PriorityQueue<>(Comparator.comparing(i -> d2[i]));
        
        Arrays.fill(d2, Long.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 (int y = 0; y < n; y++) {
                    int v = n * 1 + y;
                    if (costs[x][y] != -1 && d2[u] + costs[x][y] < d2[v]) {
                        d2[v] = d2[u] + costs[x][y];
                        q2.add(v);
                    }
                }
            } else {
                for (int y = 0; y < n; y++) {
                    if (costs[x][y] == -1) {
                        continue;
                    }
                    
                    int v1 = n * 0 + y;
                    if (d2[u] + costs[x][y] < d2[v1]) {
                        d2[v1] = d2[u] + d2[u] + costs[x][y];
                        q2.add(v1);
                    }
                    
                    int v2 = n * 1 + y;
                    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