結果
問題 | No.1812 Uribo Road |
ユーザー |
![]() |
提出日時 | 2022-01-28 01:04:28 |
言語 | Java (openjdk 23) |
結果 |
AC
|
実行時間 | 543 ms / 5,000 ms |
コード長 | 5,314 bytes |
コンパイル時間 | 5,399 ms |
コンパイル使用メモリ | 84,016 KB |
実行使用メモリ | 62,864 KB |
最終ジャッジ日時 | 2024-12-26 12:21:26 |
合計ジャッジ時間 | 14,532 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 30 |
ソースコード
import java.util.*; import java.util.stream.*; import java.io.*; class Path { int to; long d; Path(int to ,long d){ this.to = to; this.d = d; } } class Node implements Comparable<Node> { int v; long dist; Node(int v, long dist){ this.v = v; this.dist = dist; } @Override public int compareTo(Node other){ if(this.dist > other.dist) return 1; else if(this.dist < other.dist) return -1; else return 0; } } class Edge { int a, b; long c; Edge(int a, int b, long c){ this.a = a; this.b = b; this.c = c; } } public class Main { static final int INF = 1<<30; static final long INFL = 1L<<59; static final int MOD = 1000000007; static final int MOD2 = 998244353; static List<List<Path>> g; static int n; static Long[] dijkstra(int from){ Long[] ret = new Long[n]; Arrays.fill(ret, INFL); ret[from] = 0L; Queue<Node> pq = new PriorityQueue<>(); pq.add(new Node(from, 0)); while(!pq.isEmpty()){ Node node = pq.poll(); if(node.dist > ret[node.v]) continue; for(Path path : g.get(node.v)){ if(node.dist + path.d < ret[path.to]){ ret[path.to] = node.dist + path.d; pq.add(new Node(path.to, ret[path.to])); } } } return ret; } public static void main(String[] args) { FastScanner fs = new FastScanner(); PrintWriter pw = new PrintWriter(System.out); n = fs.nextInt(); int m = fs.nextInt(); int k = fs.nextInt(); int[] r = new int[k]; for(int i = 0; i < k; i++) r[i] = fs.nextInt() - 1; g = new ArrayList<>(); for(int i = 0; i < n; i++) g.add(new ArrayList<>()); Edge[] edge = new Edge[m]; for(int i = 0; i < m; i++){ int a = fs.nextInt() - 1; int b = fs.nextInt() - 1; long d = fs.nextLong(); g.get(a).add(new Path(b, d)); g.get(b).add(new Path(a, d)); edge[i] = new Edge(a, b, d); } Set<Integer> set = new HashSet<>(); set.add(0); set.add(n-1); for(int i = 0; i < k; i++){ set.add(edge[r[i]].a); set.add(edge[r[i]].b); } Map<Integer, Long[]> map = new HashMap<>(); for(int v : set){ Long[] t = dijkstra(v); map.put(v, t); } long[][][] dp = new long[1<<k][k][2]; for(int i = 0; i < 1<<k; i++){ for(int j = 0; j < k; j++){ dp[i][j][0] = INFL; dp[i][j][1] = INFL; } } for(int i = 0; i < k; i++){ dp[1<<i][i][0] = map.get(0)[edge[r[i]].b] + edge[r[i]].c; dp[1<<i][i][1] = map.get(0)[edge[r[i]].a] + edge[r[i]].c; } for(int s = 0; s < 1<<k; s++){ for(int i = 0; i < k; i++){ if(s != 0 && (s & 1<<i) == 0) continue; int a = edge[r[i]].a; int b = edge[r[i]].b; for(int j = 0; j < k; j++){ if((s & 1<<j) != 0) continue; int na = edge[r[j]].a; int nb = edge[r[j]].b; long nc = edge[r[j]].c; dp[s|1<<j][j][0] = Math.min(dp[s|1<<j][j][0], dp[s][i][0] + map.get(a)[nb] + nc); dp[s|1<<j][j][0] = Math.min(dp[s|1<<j][j][0], dp[s][i][1] + map.get(b)[nb] + nc); dp[s|1<<j][j][1] = Math.min(dp[s|1<<j][j][1], dp[s][i][0] + map.get(a)[na] + nc); dp[s|1<<j][j][1] = Math.min(dp[s|1<<j][j][1], dp[s][i][1] + map.get(b)[na] + nc); } } } long ans = INFL; for(int i = 0; i < k; i++){ int a = edge[r[i]].a; int b = edge[r[i]].b; ans = Math.min(ans, dp[(1<<k)-1][i][0] + map.get(a)[n-1]); ans = Math.min(ans, dp[(1<<k)-1][i][1] + map.get(b)[n-1]); } pw.println(ans); pw.close(); } } class FastScanner { private InputStream in = System.in; private byte[] buffer = new byte[1024]; private int length = 0, p = 0; private boolean hasNextByte() { if (p < length) return true; else{ p = 0; try{ length = in.read(buffer); }catch(Exception e){ e.printStackTrace(); } if(length == 0) return false; } return true; } private int readByte() { if (hasNextByte() == true) return buffer[p++]; return -1; } private static boolean isPrintable(int n) { return 33 <= n && n <= 126; } private void skip() { while (hasNextByte() && !isPrintable(buffer[p])) p++; } public boolean hasNext() { skip(); return hasNextByte(); } public String next() { if(!hasNext()) throw new NoSuchElementException(); StringBuilder sb = new StringBuilder(); int temp = readByte(); while (isPrintable(temp)) { sb.appendCodePoint(temp); temp = readByte(); } return sb.toString(); } public int nextInt() { return Math.toIntExact(nextLong()); } public int[] nextInts(int n) { int[] ar = new int[n]; for (int i = 0; i < n; i++) ar[i] = nextInt(); return ar; } public long[] nextLongs(int n) { long[] ar = new long[n]; for (int i = 0; i < n; i++) ar[i] = nextLong(); return ar; } public long nextLong() { if(!hasNext()) throw new NoSuchElementException(); boolean minus = false; int temp = readByte(); if (temp == '-') { minus = true; temp = readByte(); } if (temp < '0' || '9' < temp) throw new NumberFormatException(); long n = 0; while (isPrintable(temp)) { if ('0' <= temp && temp <= '9') { n *= 10; n += temp - '0'; } temp = readByte(); } return minus ? -n : n; } public char[] nextChars() { String s = next(); return s.toCharArray(); } public char[][] nextCharMat(int h, int w) { char[][] ar = new char[h][w]; for(int i = 0; i < h; i++){ String s = next(); for(int j = 0; j < w; j++){ ar[i][j] = s.charAt(j); } } return ar; } }