結果

問題 No.1797 永遠のグリッド
ユーザー H3PO4
提出日時 2022-01-01 01:03:55
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 283 ms / 2,000 ms
コード長 867 bytes
コンパイル時間 296 ms
コンパイル使用メモリ 82,520 KB
実行使用メモリ 77,060 KB
最終ジャッジ日時 2024-10-09 11:06:16
合計ジャッジ時間 3,524 ms
ジャッジサーバーID
(参考情報)
judge5 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 27
権限があれば一括ダウンロードができます

ソースコード

diff #

H, W, K = map(int, input().split())

st = set()
for b in range(K ** (H * W)):
    ct = set((b // (K ** i)) % K for i in range(H * W))
    if len(ct) != K:
        continue

    bh = b
    flag = False
    for h in range(H):
        bw = bh
        for w in range(W):
            # print(h, w, bw)
            # print(np.array([(bw // (K ** i)) % K for i in range(H * W)]).reshape((H, W)))
            if bw in st:
                flag = True
                break

            bw *= K
            for h2 in range(H):
                m = (bw // (K ** ((h2 + 1) * W))) % K
                bw += m * (K ** (h2 * W) - K ** ((h2 + 1) * W))

        if flag:
            break
        bh *= (K ** W)
        for w2 in range(W):
            m = (bh // (K ** (H * W + w2))) % K
            bh += m * (K ** w2 - K ** (H * W + w2))

    else:
        st.add(b)

print(len(st))
0