結果

問題 No.2983 Christmas Color Grid (Advent Calender ver.)
ユーザー hirayuu_yc
提出日時 2024-11-20 14:57:27
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 2,546 ms / 3,340 ms
コード長 1,006 bytes
コンパイル時間 387 ms
コンパイル使用メモリ 81,920 KB
実行使用メモリ 80,000 KB
最終ジャッジ日時 2024-11-20 18:25:23
合計ジャッジ時間 17,420 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 64
権限があれば一括ダウンロードができます

ソースコード

diff # 

four=((0,1),(1,0),(0,-1),(-1,0))
cnt=0
def pos(x,y):
    return x*W+y
def dfs(i1,i2):
    global cnt
    ad=[]
    nc=0
    for i in range(H):
        for j in range(W):
            if (i1>>pos(i,j))&1:
                nc+=1
                continue
            flg=False
            for dx,dy in four:
                nx=i+dx
                ny=j+dy
                if nx<0 or H<=nx or ny<0 or W<=ny:
                    continue
                if (i1>>pos(nx,ny))&1:
                    nc+=1
                    flg=True
                    break
            if (i2>>pos(i,j))&1:
                continue
            if flg:
                i2+=1<<pos(i,j)
                dfs(i1+(1<<pos(i,j)),i2)
    cnt+=rest[H*W-nc]*sz[i1.bit_count()]
H,W,K,M=map(int,input().split())
rest=[0]*(H*W+1)
sz=[0]*(H*W+1)
for i in range(H*W+1):
	rest[i]=pow(2,i,M)
	sz[i]=pow(i,K,M)
for i in range(H*W):
    dfs(1<<i,(1<<(i+1))-1)
ans=0
for i in range(H*W):
    ans+=cnt*pow(2,-H*W,M)*pow(i+1,-1,M)
    ans%=M
print(ans)
0