結果

問題 No.1797 永遠のグリッド
ユーザー titia
提出日時 2022-01-01 01:12:49
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 1,663 ms / 2,000 ms
コード長 1,038 bytes
コンパイル時間 422 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 23,256 KB
最終ジャッジ日時 2024-10-09 11:14:21
合計ジャッジ時間 8,918 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 27
権限があれば一括ダウンロードができます

ソースコード

diff # 

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