結果

問題 No.2159 Filling 4x4 array
ユーザー vwxyz
提出日時 2024-05-03 14:06:52
言語 PyPy3
(7.3.15)
結果
MLE  
実行時間 -
コード長 1,206 bytes
コンパイル時間 229 ms
コンパイル使用メモリ 81,920 KB
実行使用メモリ 251,020 KB
最終ジャッジ日時 2024-11-24 14:39:46
合計ジャッジ時間 138,421 ms
ジャッジサーバーID
(参考情報)
judge2 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2 MLE * 3
other AC * 8 TLE * 20 MLE * 17
権限があれば一括ダウンロードができます

ソースコード

diff #

import sys
readline=sys.stdin.readline
from functools import lru_cache
from collections import defaultdict

HW=list(map(int,readline().split()))
for i in range(8):
    HW[i]-=4
H,W=HW[:4],HW[4:]
H.sort()
W.sort()
mod=998244353
cnt=[defaultdict(int) for c in range(390625)]
se=set()
for bit in range(1<<16):
    HH=tuple(sum(bit>>h*4+w&1 for w in range(4)) for h in range(4))
    WW=tuple(sum(bit>>h*4+w&1 for h in range(4)) for w in range(4))
    cnt[sum(HH[i]%2*5**i for i in range(4))+sum(WW[i]%2*5**(i+4) for i in range(4))][(HH,WW)]+=1
    se.add((HH,WW))
@lru_cache(maxsize=None)
def solve(H,W):
    for h in H:
        if h<0:
            return 0
    for w in W:
        if w<0:
            return 0
    if all(h==0 for h in H) and all(w==0 for w in W):
        return 1
    retu=0
    for (HH,WW),c in cnt[sum(H[i]%2*5**i for i in range(4))+sum(W[i]%2*5**(i+4) for i in range(4))].items():
        if all(h>=hh and h%2==hh%2 for h,hh in zip(H,HH)) and all(w>=ww and w%2==ww%2 for w,ww in zip(W,WW)):
            retu+=solve(tuple(sorted([h-hh>>1 for h,hh in zip(H,HH)])),tuple(sorted([w-ww>>1 for w,ww in zip(W,WW)])))*c
            retu%=mod
    return retu
ans=solve(tuple(H),tuple(W))
print(ans)
0