結果

問題 No.1690 Power Grid
コンテスト
ユーザー brthyyjp
提出日時 2021-09-24 23:06:55
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 1,324 bytes
コンパイル時間 262 ms
コンパイル使用メモリ 81,980 KB
実行使用メモリ 99,508 KB
最終ジャッジ日時 2024-07-05 11:23:19
合計ジャッジ時間 4,976 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 3 TLE * 1 -- * 21
権限があれば一括ダウンロードができます

ソースコード

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)
    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)
                U = []
                for u in range(n):
                    if (s>>u)&1:
                        U.append(u)
                if not U:
                    nx[ns] = min(nx[ns], dp[s]+A[v])
                else:
                    d = INF
                    for u in U:
                        d = min(d, D[v][u])
                    nx[s|ns] = min(nx[ns], dp[s]+A[v]+d)
        dp = nx
    print(min(dp))

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