結果

問題 No.1690 Power Grid
ユーザー brthyyjpbrthyyjp
提出日時 2021-09-24 23:11:23
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 2,314 ms / 3,000 ms
コード長 1,394 bytes
コンパイル時間 161 ms
コンパイル使用メモリ 82,688 KB
実行使用メモリ 171,776 KB
最終ジャッジ日時 2024-07-05 11:28:53
合計ジャッジ時間 29,933 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 39 ms
52,224 KB
testcase_01 AC 42 ms
52,096 KB
testcase_02 AC 75 ms
69,120 KB
testcase_03 AC 40 ms
52,224 KB
testcase_04 AC 40 ms
52,480 KB
testcase_05 AC 40 ms
52,224 KB
testcase_06 AC 1,481 ms
153,216 KB
testcase_07 AC 1,473 ms
153,344 KB
testcase_08 AC 1,482 ms
153,088 KB
testcase_09 AC 1,482 ms
153,088 KB
testcase_10 AC 773 ms
138,880 KB
testcase_11 AC 2,117 ms
169,344 KB
testcase_12 AC 41 ms
52,480 KB
testcase_13 AC 76 ms
71,424 KB
testcase_14 AC 220 ms
84,608 KB
testcase_15 AC 2,314 ms
171,648 KB
testcase_16 AC 1,522 ms
153,344 KB
testcase_17 AC 826 ms
114,560 KB
testcase_18 AC 418 ms
94,336 KB
testcase_19 AC 1,515 ms
152,960 KB
testcase_20 AC 1,560 ms
153,344 KB
testcase_21 AC 891 ms
136,448 KB
testcase_22 AC 1,798 ms
161,920 KB
testcase_23 AC 2,237 ms
171,776 KB
testcase_24 AC 2,051 ms
165,504 KB
testcase_25 AC 2,095 ms
167,680 KB
testcase_26 AC 2,172 ms
169,728 KB
testcase_27 AC 41 ms
52,096 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

INF = 10**18

def main():
    n, m, k = map(int, input().split())
    A = list(map(int, input().split()))
    g = [[] for i in range(n)]
    D = [[INF]*n for i in range(n)]
    for i in range(m):
        x, y, z = map(int, input().split())
        x, y = x-1, y-1
        g[x].append((z, y))
        g[y].append((z, x))
        D[x][y] = z
        D[y][x] = z

    for i in range(n):
        D[i][i] = 0

    def warshal_floyd(d):
        for k in range(n):
            for i in range(n):
                for j in range(n):
                    d[i][j] = min(d[i][j], d[i][k]+d[k][j])
        return d

    D = warshal_floyd(D)
    mn = [[INF]*n for i in range(1<<n)]
    for s in range(1<<n):
        U = []
        for u in range(n):
            if (s>>u)&1:
                U.append(u)
        for v in range(n):
            d = INF
            for u in U:
                d = min(d, D[v][u])
            mn[s][v] = d
    dp = [INF]*(1<<n)
    dp[0] = 0
    for i in range(k):
        nx = [INF]*(1<<n)
        for s in range(1<<n):
            for v in range(n):
                if (s>>v)&1:
                    continue
                ns = s|(1<<v)
                if s == 0:
                    nx[ns] = min(nx[ns], dp[s]+A[v])
                else:
                    nx[s|ns] = min(nx[ns], dp[s]+A[v]+mn[s][v])
        dp = nx
    print(min(dp))

if __name__ == '__main__':
    main()
0