結果

問題 No.95 Alice and Graph
ユーザー しらっ亭しらっ亭
提出日時 2015-10-01 01:27:55
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,381 bytes
コンパイル時間 173 ms
コンパイル使用メモリ 82,148 KB
実行使用メモリ 95,872 KB
最終ジャッジ日時 2024-07-19 18:46:47
合計ジャッジ時間 5,611 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 WA -
testcase_01 AC 295 ms
95,192 KB
testcase_02 AC 282 ms
95,872 KB
testcase_03 AC 84 ms
82,304 KB
testcase_04 WA -
testcase_05 WA -
testcase_06 WA -
testcase_07 AC 286 ms
95,548 KB
testcase_08 AC 65 ms
69,376 KB
testcase_09 AC 37 ms
52,340 KB
testcase_10 AC 41 ms
58,368 KB
testcase_11 AC 69 ms
69,504 KB
testcase_12 AC 64 ms
66,944 KB
testcase_13 AC 2,171 ms
92,508 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

# めぐるちゃんのを参考にしました

def solve():
    N, M, K = map(int, input().split())

    dist = [[float('inf')] * N for _ in range(N)]
    for i in range(N):
        dist[i][i] = 0

    for i in range(M):
        u, v = map(int, input().split())
        u -= 1
        v -= 1
        dist[u][v] = dist[v][u] = 1

    for k in range(N):
        for i in range(N):
            for j in range(N):
                dist[i][j] = min(dist[i][j], dist[i][k] + dist[k][j])

    inf = float('inf')

    dp = [[inf] * (K + 1) for _ in range(1 << (K + 1))]

    def check(v):
        l = len(v)

        for i in range(l):
            dp[1 << i][i] = dist[0][v[i]]

        for i in range(1, 1 << l):
            dpi = dp[i]
            for j in range(l):
                if dpi[j] == inf:
                    continue
                for k in range(l):
                    k1 = 1 << k
                    if i & k1:
                        continue
                    ni = i | k1
                    dp[ni][k] = min(dp[ni][k], dpi[j] + dist[v[j]][v[k]])

        return any(a <= K for a in dp[1 << l - 1][:K])

    ans = 0
    v = []
    for n in range(N - 1, 0, -1):
        if len(v) == K:
            break
        v.append(n)

        if check(v):
            ans += 2 ** n - 1
        else:
            v.pop()

    print(ans)


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