結果
| 問題 | No.2164 Equal Balls |
| ユーザー |
 titia
|
| 提出日時 | 2022-12-15 09:25:53 |
| 言語 | PyPy3 (7.3.15) |
| 結果 |
TLE
|
| 実行時間 | - |
| コード長 | 1,217 bytes |
| コンパイル時間 | 331 ms |
| コンパイル使用メモリ | 82,304 KB |
| 実行使用メモリ | 173,604 KB |
| 最終ジャッジ日時 | 2024-11-09 02:46:48 |
| 合計ジャッジ時間 | 9,334 ms |
|
ジャッジサーバーID （参考情報） |
judge1 / judge4 |
（要ログイン）
テストケース
テストケース表示| 入力 | 結果 | 実行時間 実行使用メモリ |
|---|---|---|
| testcase_00 | AC | 93 ms
97,664 KB |
| testcase_01 | AC | 95 ms
99,200 KB |
| testcase_02 | AC | 120 ms
108,544 KB |
| testcase_03 | AC | 98 ms
99,328 KB |
| testcase_04 | AC | 103 ms
103,168 KB |
| testcase_05 | AC | 105 ms
105,728 KB |
| testcase_06 | AC | 109 ms
106,368 KB |
| testcase_07 | AC | 104 ms
103,680 KB |
| testcase_08 | TLE | - |
| testcase_09 | -- | - |
| testcase_10 | -- | - |
| testcase_11 | -- | - |
| testcase_12 | -- | - |
| testcase_13 | -- | - |
| testcase_14 | -- | - |
| testcase_15 | -- | - |
| testcase_16 | -- | - |
| testcase_17 | -- | - |
| testcase_18 | -- | - |
| testcase_19 | -- | - |
| testcase_20 | -- | - |
| testcase_21 | -- | - |
| testcase_22 | -- | - |
| testcase_23 | -- | - |
| testcase_24 | -- | - |
| testcase_25 | -- | - |
| testcase_26 | -- | - |
| testcase_27 | -- | - |
| testcase_28 | -- | - |
| testcase_29 | -- | - |
| testcase_30 | -- | - |
| testcase_31 | -- | - |
| testcase_32 | -- | - |
| testcase_33 | -- | - |
| testcase_34 | -- | - |
| testcase_35 | -- | - |
| testcase_36 | -- | - |
| testcase_37 | -- | - |
| testcase_38 | -- | - |
| testcase_39 | -- | - |
| testcase_40 | -- | - |
| testcase_41 | -- | - |
| testcase_42 | -- | - |
| testcase_43 | -- | - |
| testcase_44 | -- | - |
| testcase_45 | -- | - |
| testcase_46 | -- | - |
| testcase_47 | -- | - |
| testcase_48 | -- | - |
| testcase_49 | -- | - |
| testcase_50 | -- | - |
| testcase_51 | -- | - |
| testcase_52 | -- | - |
| testcase_53 | -- | - |
ソースコード
import sys
input = sys.stdin.readline
N,M=map(int,input().split())
A=list(map(int,input().split()))
B=list(map(int,input().split()))
mod=998244353
FACT=[1]
for i in range(1,2*10**5+1):
FACT.append(FACT[-1]*i%mod)
FACT_INV=[pow(FACT[-1],mod-2,mod)]
for i in range(2*10**5,0,-1):
FACT_INV.append(FACT_INV[-1]*i%mod)
FACT_INV.reverse()
def Combi(a,b):
if 0<=b<=a:
return FACT[a]*FACT_INV[b]%mod*FACT_INV[a-b]%mod
else:
return 0
PLUS=[[1]*601 for i in range(M)]
for i in range(N):
k=i%M
a=A[i]
b=B[i]
for j in range(601):
p=j-300
# bの方がaよりp多い
score=0
for x in range(a+1):
y=x+p
if 0<=y<=b:
score+=Combi(a,x)*Combi(b,y)
score%=mod
PLUS[k][j]=PLUS[k][j]*score%mod
DP=[0]*(2*300*300+1)
DP[300*300]=1
for i in range(M):
NDP=[0]*(2*300*300+1)
for j in range(2*300*300+1):
if DP[j]==0:
continue
for k in range(601):
p=k-300
if 0<=j+p<2*300*300:
NDP[j+p]=(NDP[j+p]+DP[j]*PLUS[i][k])%mod
DP=NDP
print(DP[300*300])