結果

問題 No.1797 永遠のグリッド
ユーザー titia
提出日時 2022-01-01 01:10:18
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 987 ms / 2,000 ms
コード長 1,077 bytes
コンパイル時間 450 ms
コンパイル使用メモリ 82,176 KB
実行使用メモリ 95,252 KB
最終ジャッジ日時 2024-10-09 11:12:06
合計ジャッジ時間 6,353 ms
ジャッジサーバーID
(参考情報)
judge4 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
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]

                for j in range(H):
                    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