結果

問題 No.2159 Filling 4x4 array
ユーザー vwxyz
提出日時 2023-02-16 08:06:03
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 1,116 bytes
コンパイル時間 986 ms
コンパイル使用メモリ 82,232 KB
実行使用メモリ 116,968 KB
最終ジャッジ日時 2024-07-18 10:51:01
合計ジャッジ時間 17,045 ms
ジャッジサーバーID
(参考情報)
judge3 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 5
other AC * 20 TLE * 1 -- * 24
権限があれば一括ダウンロードができます

ソースコード

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(lambda:defaultdict(int))
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[(tuple(h%2 for h in HH),tuple(w%2 for w in WW))][(HH,WW)]+=1
@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[(tuple(h%2 for h in H),tuple(w%2 for w in W))].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