結果

問題 No.2983 Christmas Color Grid (Advent Calender ver.)
コンテスト
ユーザー rin204
提出日時 2024-12-08 00:57:15
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 2,202 ms / 3,340 ms
コード長 1,311 bytes
コンパイル時間 360 ms
コンパイル使用メモリ 81,920 KB
実行使用メモリ 82,576 KB
最終ジャッジ日時 2024-12-08 07:26:18
合計ジャッジ時間 16,752 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 64
権限があれば一括ダウンロードができます

ソースコード

diff # 

h, w, K, MOD = map(int, input().split())

hw = h * w
powK = [pow(i, K, MOD) for i in range(hw + 1)]
pow2 = [1] * (hw + 1)
for i in range(1, hw + 1):
    pow2[i] = pow2[i - 1] * 2 % MOD

A = [[-1] * w for _ in range(h)]
didj = [(0, 1), (1, 0), (0, -1), (-1, 0)]
ans = 0


def sol(c1, c2, si, sj):
    ii = -1
    jj = -1
    for i in range(h):
        for j in range(w):
            if A[i][j] != -1:
                continue

            for di, dj in didj:
                ni = i + di
                nj = j + dj
                if ni < 0 or ni >= h or nj < 0 or nj >= w:
                    continue
                if A[ni][nj] == 1:
                    ii = i
                    jj = j
                    break

        if ii != -1:
            break

    if ii == -1:
        global ans
        ans += powK[c1] * pow2[h * w - c2] % MOD
    else:
        if si * w + sj < ii * w + jj:
            A[ii][jj] = 1
            sol(c1 + 1, c2 + 1, si, sj)
        A[ii][jj] = 0
        sol(c1, c2 + 1, si, sj)
        A[ii][jj] = -1


for si in range(h):
    for sj in range(w):
        A[si][sj] = 1
        sol(1, 1, si, sj)
        A[si][sj] = -1

ans %= MOD
ans *= pow(pow2[h * w], MOD - 2, MOD)
ans %= MOD
inv = 0
for i in range(1, h * w + 1):
    inv += pow(i, MOD - 2, MOD)

ans *= inv
print(ans % MOD)
0