結果

問題 No.1690 Power Grid
ユーザー nebukuro09nebukuro09
提出日時 2021-09-24 22:22:19
言語 D
(dmd 2.106.1)
結果
AC  
実行時間 877 ms / 3,000 ms
コード長 1,618 bytes
コンパイル時間 884 ms
コンパイル使用メモリ 121,516 KB
実行使用メモリ 84,056 KB
最終ジャッジ日時 2024-06-22 12:36:35
合計ジャッジ時間 13,150 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
6,812 KB
testcase_01 AC 1 ms
6,816 KB
testcase_02 AC 2 ms
6,940 KB
testcase_03 AC 1 ms
6,940 KB
testcase_04 AC 1 ms
6,940 KB
testcase_05 AC 1 ms
6,944 KB
testcase_06 AC 593 ms
64,148 KB
testcase_07 AC 589 ms
62,896 KB
testcase_08 AC 591 ms
66,144 KB
testcase_09 AC 591 ms
65,436 KB
testcase_10 AC 413 ms
48,356 KB
testcase_11 AC 806 ms
82,596 KB
testcase_12 AC 2 ms
6,940 KB
testcase_13 AC 3 ms
6,940 KB
testcase_14 AC 67 ms
9,180 KB
testcase_15 AC 877 ms
84,056 KB
testcase_16 AC 622 ms
63,500 KB
testcase_17 AC 297 ms
30,240 KB
testcase_18 AC 134 ms
15,700 KB
testcase_19 AC 618 ms
63,656 KB
testcase_20 AC 634 ms
65,900 KB
testcase_21 AC 429 ms
45,484 KB
testcase_22 AC 729 ms
72,096 KB
testcase_23 AC 873 ms
82,224 KB
testcase_24 AC 792 ms
75,628 KB
testcase_25 AC 829 ms
77,132 KB
testcase_26 AC 842 ms
79,536 KB
testcase_27 AC 1 ms
6,944 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import std.stdio, std.array, std.string, std.conv, std.algorithm;
import std.typecons, std.range, std.random, std.math, std.container;
import std.numeric, std.bigint, core.bitop;

immutable long MOD = 998244353;
immutable long inf = 1L << 59;

void main() {
    auto s = readln.split.map!(to!int);
    auto N = s[0];
    auto M = s[1];
    auto K = s[2];
    auto A = readln.split.map!(to!long).array;
    auto G = new long[][](N, N);
    foreach (i; 0..N) G[i][] = inf;
    foreach (i; 0..M) {
        s = readln.split.map!(to!int);
        auto u = s[0] - 1;
        auto v = s[1] - 1;
        auto z = s[2];
        G[u][v] = G[v][u] = z.to!long;
    }

    auto dist = new long[][](N, N);
    foreach (i; 0..N) foreach (j; 0..N) dist[i][j] = G[i][j];

    foreach (i; 0..N) foreach (j; 0..N) foreach (k; 0..N) if (dist[j][i] != inf && dist[i][k] != inf) {
        dist[j][k] = min(dist[j][k], dist[j][i] + dist[i][k]);
    }

    auto dist2 = new long[][](N, 1<<N);
    foreach (i; 0..N) dist2[i][] = inf;
    foreach (i; 0..N) foreach (mask; 0..1<<N) {
        foreach (j; 0..N) if (mask & (1 << j)) dist2[i][mask] = min(dist2[i][mask], dist[i][j]);
    }

    auto dp = new long[][](K+1, 1<<N);
    foreach (i; 0..K+1) dp[i][] = inf;
    dp[0][0] = 0;

    foreach (i; 0..K) foreach (mask; 0..1<<N) foreach (j; 0..N) if (!(mask & (1 << j))) {
        if (i == 0) {
            dp[i+1][mask|(1<<j)] = min(dp[i+1][mask|(1<<j)], dp[i][mask] + A[j]);
        } else {
            dp[i+1][mask|(1<<j)] = min(dp[i+1][mask|(1<<j)], dp[i][mask] + A[j] + dist2[j][mask]);
        }
    }

    dp[K].reduce!min.writeln;
}
0