結果

問題 No.1797 永遠のグリッド
ユーザー titia
提出日時 2022-01-01 01:12:49
言語 Python3
(3.14.3 + numpy 2.4.4 + scipy 1.17.1)
コンパイル:
python3 -mpy_compile _filename_
実行:
python3 _filename_
結果
AC  
実行時間 1,042 ms / 2,000 ms
コード長 1,038 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 631 ms
コンパイル使用メモリ 20,568 KB
実行使用メモリ 27,896 KB
最終ジャッジ日時 2026-04-25 14:27:50
合計ジャッジ時間 10,067 ms
ジャッジサーバーID
(参考情報)
judge2_0 / judge1_1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 27
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code

from copy import deepcopy

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

def calc(H,W,K):
    if K==1:
        return 1

    SET=set()
    ANS=0

    A=[[0]*W for i in range(H)]

    for i in range(K**(H*W)):
        x=i
        for j in range(H):
            for k in range(W):
                A[j][k]=x%K
                x//=K

        B=deepcopy(A)

        for i in range(H):
            B[i]=tuple(B[i])

        flag=1

        for i in range(H):
            if tuple(B[i:]+B[:i]) in SET:
                flag=0
                break

        for i in range(W):
            for j in range(H):
                B[j]=B[j][1:]+B[j][:1]

            if tuple(B) in SET:
                flag=0
                break

        if flag==1:
            ANS+=1

            for i in range(W):
                for j in range(H):
                    B[j]=B[j][1:]+B[j][:1]

                SET.add(tuple(B[i:]+B[:i]))
    return ANS

if K==1:
    print(1)
elif K==2:
    print(calc(H,W,K)-2)
else:
    print(calc(H,W,3)-3*calc(H,W,2)+3*calc(H,W,1))
    
    
0