結果
問題 |
No.788 トラックの移動
|
ユーザー |
![]() |
提出日時 | 2020-05-14 19:54:37 |
言語 | Java (openjdk 23) |
結果 |
WA
|
実行時間 | - |
コード長 | 2,061 bytes |
コンパイル時間 | 2,349 ms |
コンパイル使用メモリ | 79,540 KB |
実行使用メモリ | 72,608 KB |
最終ジャッジ日時 | 2024-09-15 19:07:46 |
合計ジャッジ時間 | 14,339 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge6 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 11 WA * 3 |
ソースコード
import java.util.*; public class Main { public static void main (String[] args) { Scanner sc = new Scanner(System.in); int n = sc.nextInt(); int m = sc.nextInt(); int l = sc.nextInt() - 1; int[] trucks = new int[n]; ArrayList<HashMap<Integer, Integer>> graph = new ArrayList<>(); int count = 0; int[][] costs = new int[n][n]; for (int i = 0; i < n; i++) { trucks[i] = sc.nextInt(); if (trucks[i] > 0) { count++; } graph.add(new HashMap<>()); Arrays.fill(costs[i], Integer.MAX_VALUE); } if (count <= 1) { System.out.println(0); return; } for (int i = 0 ; i < m; i++) { int a = sc.nextInt() - 1; int b = sc.nextInt() - 1; int c = sc.nextInt(); graph.get(a).put(b, c); graph.get(b).put(a, c); } PriorityQueue<Path> queue = new PriorityQueue<>(); for (int i = 0; i < n; i++) { queue.add(new Path(i, 0)); while (queue.size() > 0) { Path p = queue.poll(); if (costs[i][p.idx] <= p.value) { continue; } costs[i][p.idx] = p.value; for (Map.Entry<Integer, Integer> entry : graph.get(p.idx).entrySet()) { if (costs[i][entry.getKey()] == Integer.MAX_VALUE) { queue.add(new Path(entry.getKey(), entry.getValue() + p.value)); } } } } long min = Long.MAX_VALUE; for (int i = 0; i < n; i++) { long max = Long.MIN_VALUE; long total = 0; for (int j = 0; j < n; j++) { if (trucks[j] > 0) { total += costs[i][j] * (long)trucks[j] * 2; max = Math.max(max, costs[i][j] - costs[j][l]); } } min = Math.min(min, total - max); } System.out.println(min); } static class Path implements Comparable<Path> { int idx; int value; public Path (int idx, int value) { this.idx = idx; this.value = value; } public int compareTo(Path another) { return value - another.value; } } }