結果
問題 | No.788 トラックの移動 |
ユーザー | hiromi_ayase |
提出日時 | 2019-02-08 23:23:28 |
言語 | Java21 (openjdk 21) |
結果 |
AC
|
実行時間 | 1,609 ms / 2,000 ms |
コード長 | 4,930 bytes |
コンパイル時間 | 2,480 ms |
コンパイル使用メモリ | 83,200 KB |
実行使用メモリ | 51,156 KB |
最終ジャッジ日時 | 2024-05-08 22:57:15 |
合計ジャッジ時間 | 13,185 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge2 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 1,584 ms
50,800 KB |
testcase_01 | AC | 74 ms
38,128 KB |
testcase_02 | AC | 76 ms
38,128 KB |
testcase_03 | AC | 76 ms
37,984 KB |
testcase_04 | AC | 725 ms
50,732 KB |
testcase_05 | AC | 1,609 ms
51,156 KB |
testcase_06 | AC | 1,580 ms
51,104 KB |
testcase_07 | AC | 74 ms
37,964 KB |
testcase_08 | AC | 75 ms
38,468 KB |
testcase_09 | AC | 73 ms
38,208 KB |
testcase_10 | AC | 75 ms
38,008 KB |
testcase_11 | AC | 78 ms
38,080 KB |
testcase_12 | AC | 77 ms
38,416 KB |
testcase_13 | AC | 73 ms
38,224 KB |
testcase_14 | AC | 77 ms
38,428 KB |
testcase_15 | AC | 636 ms
50,076 KB |
testcase_16 | AC | 1,516 ms
50,676 KB |
ソースコード
import java.util.Arrays; import java.util.Comparator; import java.util.TreeSet; public class Main { private static void solve() { int n = ni(); int m = ni(); int l = ni() - 1; int[] t = na(n); int cnt = 0; for (int i = 0; i < n; i ++) { if (t[i] > 0) cnt ++; } if (cnt <= 1) { System.out.println(0); return; } int[] from = new int[m]; int[] to = new int[m]; int[] w = new int[m]; for (int i = 0; i < m; i ++) { from[i] = ni() - 1; to[i] = ni() - 1; w[i] = ni(); } int[][][] g = packWU(n, from, to, w); long[] dl = dijk(g, l); long ret = Long.MAX_VALUE; for (int i = 0; i < n; i ++) { long[] d = dijk(g, i); long now = 0; for (int j = 0; j < n; j ++) { now += (long)d[j] * 2 * t[j]; } for (int j = 0; j < n; j ++) { if (t[j] > 0) { long now2 = now + dl[j] - d[j]; ret = Math.min(now2, ret); } } } System.out.println(ret); } public static long[] dijk(int[][][] g, int from) { int n = g.length; final long[] td = new long[n]; Arrays.fill(td, Long.MAX_VALUE / 2); TreeSet<Integer> q = new TreeSet<Integer>(new Comparator<Integer>(){ public int compare(Integer a, Integer b) { if(td[a] - td[b] != 0)return Long.signum(td[a] - td[b]); return a - b; } }); q.add(from); td[from] = 0; while(q.size() > 0){ int cur = q.pollFirst(); for(int i = 0;i < g[cur].length;i++){ int next = g[cur][i][0]; long nd = td[cur] + g[cur][i][1]; if(nd < td[next]){ q.remove(next); td[next] = nd; q.add(next); } } } return td; } public static int[][][] packWU(int n, int[] from, int[] to, int[] w){ return packWU(n, from, to, w, from.length); } public static int[][][] packWU(int n, int[] from, int[] to, int[] w, int sup) { int[][][] g = new int[n][][]; int[] p = new int[n]; for(int i = 0;i < sup;i++)p[from[i]]++; for(int i = 0;i < sup;i++)p[to[i]]++; for(int i = 0;i < n;i++)g[i] = new int[p[i]][2]; for(int i = 0;i < sup;i++){ --p[from[i]]; g[from[i]][p[from[i]]][0] = to[i]; g[from[i]][p[from[i]]][1] = w[i]; --p[to[i]]; g[to[i]][p[to[i]]][0] = from[i]; g[to[i]][p[to[i]]][1] = w[i]; } return g; } public static void main(String[] args) { new Thread(null, new Runnable() { @Override public void run() { long start = System.currentTimeMillis(); String debug = args.length > 0 ? args[0] : null; if (debug != null) { try { is = java.nio.file.Files.newInputStream(java.nio.file.Paths.get(debug)); } catch (Exception e) { throw new RuntimeException(e); } } reader = new java.io.BufferedReader(new java.io.InputStreamReader(is), 32768); solve(); out.flush(); tr((System.currentTimeMillis() - start) + "ms"); } }, "", 64000000).start(); } private static java.io.InputStream is = System.in; private static java.io.PrintWriter out = new java.io.PrintWriter(System.out); private static java.util.StringTokenizer tokenizer = null; private static java.io.BufferedReader reader; public static String next() { while (tokenizer == null || !tokenizer.hasMoreTokens()) { try { tokenizer = new java.util.StringTokenizer(reader.readLine()); } catch (Exception e) { throw new RuntimeException(e); } } return tokenizer.nextToken(); } private static double nd() { return Double.parseDouble(next()); } private static long nl() { return Long.parseLong(next()); } private static int[] na(int n) { int[] a = new int[n]; for (int i = 0; i < n; i++) a[i] = ni(); return a; } private static char[] ns() { return next().toCharArray(); } private static long[] nal(int n) { long[] a = new long[n]; for (int i = 0; i < n; i++) a[i] = nl(); return a; } private static int[][] ntable(int n, int m) { int[][] table = new int[n][m]; for (int i = 0; i < n; i++) { for (int j = 0; j < m; j++) { table[i][j] = ni(); } } return table; } private static int[][] nlist(int n, int m) { int[][] table = new int[m][n]; for (int i = 0; i < n; i++) { for (int j = 0; j < m; j++) { table[j][i] = ni(); } } return table; } private static int ni() { return Integer.parseInt(next()); } private static void tr(Object... o) { if (is != System.in) System.out.println(java.util.Arrays.deepToString(o)); } }