結果

問題 No.95 Alice and Graph
ユーザー maspymaspy
提出日時 2020-05-14 17:38:50
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 1,490 bytes
コンパイル時間 332 ms
コンパイル使用メモリ 87,304 KB
実行使用メモリ 286,388 KB
最終ジャッジ日時 2023-10-13 16:17:16
合計ジャッジ時間 14,842 ms
ジャッジサーバーID
(参考情報)
judge13 / judge15
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2,518 ms
286,388 KB
testcase_01 TLE -
testcase_02 -- -
testcase_03 -- -
testcase_04 -- -
testcase_05 -- -
testcase_06 -- -
testcase_07 -- -
testcase_08 -- -
testcase_09 -- -
testcase_10 -- -
testcase_11 -- -
testcase_12 -- -
testcase_13 -- -
権限があれば一括ダウンロードができます

ソースコード

diff #

import sys
from collections import deque

read = sys.stdin.buffer.read
readline = sys.stdin.buffer.readline
readlines = sys.stdin.buffer.readlines

N, M, K = map(int, readline().split())
m = map(int, read().split())

G = [[] for _ in range(N)]
for u, v in zip(m, m):
    u -= 1
    v -= 1
    G[u].append(v)
    G[v].append(u)

def bfs(v):
    dist = [-1] * N
    q = deque([v])
    dist[v] = 0
    while q:
        v = q.popleft()
        for w in G[v]:
            if dist[w] != -1:
                continue
            dist[w] = dist[v] + 1
            q.append(w)
    return dist

dist_mat = [bfs(v) for v in range(N)]

def extract_subgraph(A):
    assert A[0] == 0
    mat = []
    for i in A:
        mat.append([dist_mat[i][j] for j in A])
    return mat

def min_hamilton_path(A):
    mat = extract_subgraph(A)
    n = len(A)
    INF = K + 1
    dp = [[INF] * n for _ in range(1 << n)]  # (部分集合, 最後の頂点)
    dp[1][0] = 0
    for s in range(1, 1 << n, 2):
        for i in range(n):
            x = dp[s][i]
            if x == INF:
                continue
            for j in range(1, n):
                if s & (1 << j):
                    continue
                t = s ^ (1 << j)
                y = x + mat[i][j]
                if dp[t][j] > y:
                    dp[t][j] = y
    return min(dp[-1])

A = [0]
for n in range(N - 1, 0, -1):
    A.append(n)
    if min_hamilton_path(A) > K:
        A.pop()

answer = sum((1 << x) - 1 for x in A)
print(answer)
0