結果

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

ソースコード

diff #

import numpy as np

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