結果

問題 No.2178 Payable Magic Items
ユーザー minimum
提出日時 2023-01-06 22:08:48
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 813 ms / 4,000 ms
コード長 832 bytes
コンパイル時間 573 ms
コンパイル使用メモリ 82,272 KB
実行使用メモリ 178,864 KB
最終ジャッジ日時 2024-12-14 17:58:50
合計ジャッジ時間 8,947 ms
ジャッジサーバーID
(参考情報)
judge5 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 23
権限があれば一括ダウンロードができます

ソースコード

diff #

from collections import deque


N, K = map(int, input().split())
S = [input() for _ in range(N)]
num = dict()

for i in range(N):
    n = 0
    for j in range(K):
        n *= 5
        n += 4 - int(S[i][j])
    num[n] = i

if 0 in num:
    print(N - 1)
    exit()

dq = deque()
dq.append((0, 0))
done = [-1] * (5 ** K)

while dq:
    now, t = dq.popleft()
    if done[now] >= t:
        continue
    done[now] = t
    if now in num:
        t = 1
    a = []
    for i in range(K):
        a.append(now % 5)
        now //= 5
    a.reverse()
    for i in range(K):
        if a[i] == 4:
            continue
        a[i] += 1
        nxt = 0
        for j in range(K):
            nxt *= 5
            nxt += a[j]
        dq.append((nxt, t))
        a[i] -= 1


ans = 0
for n in num:
    if done[n] == 1:
        ans += 1
print(ans)
0