結果

問題 No.95 Alice and Graph
ユーザー しらっ亭しらっ亭
提出日時 2015-10-01 02:06:42
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 2,214 ms / 5,000 ms
コード長 1,618 bytes
コンパイル時間 500 ms
コンパイル使用メモリ 81,836 KB
実行使用メモリ 96,000 KB
最終ジャッジ日時 2024-11-14 13:34:09
合計ジャッジ時間 11,968 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1,299 ms
95,504 KB
testcase_01 AC 303 ms
95,500 KB
testcase_02 AC 295 ms
95,560 KB
testcase_03 AC 79 ms
76,532 KB
testcase_04 AC 1,761 ms
95,736 KB
testcase_05 AC 1,220 ms
95,476 KB
testcase_06 AC 1,769 ms
95,632 KB
testcase_07 AC 301 ms
96,000 KB
testcase_08 AC 72 ms
75,476 KB
testcase_09 AC 36 ms
53,096 KB
testcase_10 AC 41 ms
59,880 KB
testcase_11 AC 74 ms
76,052 KB
testcase_12 AC 63 ms
71,728 KB
testcase_13 AC 2,214 ms
93,024 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

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


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

    inf = float('inf')

    dist = [[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, i, j in itertools.product(range(N), repeat=3):
        dikj = dist[i][k] + dist[k][j]
        if dist[i][j] > dikj:
            dist[i][j] = dikj

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

    def check(v):
        l = len(v)
        l1 = 1 << l

        for i in range(l1):
            for j in range(l):
                dp[i][j] = inf

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

        for i in range(1, l1):
            dpi = dp[i]
            for j in range(l):
                dpij = dpi[j]
                if dpij == inf:
                    continue
                distj = dist[v[j]]
                for k in range(l):
                    k1 = 1 << k
                    if i & k1:
                        continue
                    nd = dp[i + k1]
                    nv = dpij + distj[v[k]]
                    if nv < nd[k]:
                        nd[k] = nv

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

    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