結果
問題 | No.2178 Payable Magic Items |
ユーザー |
|
提出日時 | 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 |
ソースコード
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)