結果

問題 No.788 トラックの移動
コンテスト
ユーザー tenten
提出日時 2020-09-03 22:58:42
言語 Java
(openjdk 23)
結果
WA  
実行時間 -
コード長 2,054 bytes
コンパイル時間 2,301 ms
コンパイル使用メモリ 79,744 KB
実行使用メモリ 73,916 KB
最終ジャッジ日時 2024-11-24 05:15:06
合計ジャッジ時間 14,089 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 9 WA * 5
権限があれば一括ダウンロードができます

ソースコード

diff # 

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 start = sc.nextInt() - 1;
    	int[] trucks = new int[n];
    	ArrayList<HashMap<Integer, Integer>> graph = new ArrayList<>();
    	for (int i = 0; i < n; i++) {
    	    trucks[i] = sc.nextInt();
    	    graph.add(new HashMap<>());
    	}
    	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);
    	}
    	int[][] costs = new int[n][n];
    	PriorityQueue<Path> queue = new PriorityQueue<>();
    	for (int i = 0; i < n; i++) {
    	    Arrays.fill(costs[i], Integer.MAX_VALUE);
    	    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 (int x : graph.get(p.idx).keySet()) {
    	            if (costs[i][x] != Integer.MAX_VALUE) {
    	                continue;
    	            }
    	            queue.add(new Path(x, p.value + graph.get(p.idx).get(x)));
    	        }
    	    }
    	}
    	long min = Long.MAX_VALUE;
    	for (int i = 0; i < n; i++) {
    	    int other = Integer.MAX_VALUE;
    	    long base = 0;
    	    for (int j = 0; j < n; j++) {
    	        base += costs[i][j] * (long)trucks[j] * 2;
    	        if (trucks[j] != 0) {
    	            other = Math.min(other, costs[start][j] - costs[i][j]);
    	        }
    	    }
    	    min = Math.min(min, base + other);
    	}
    	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;
	    }
	}
}
0